From df6fe9854d1b97ba613e161cba477cb181325413 Mon Sep 17 00:00:00 2001
From: Yizhou Chi <chiyizhou@fuzhi.ai>
Date: Fri, 6 Sep 2024 19:27:36 +0800
Subject: [PATCH] fix dataset bug

---
 expo/data/dataset.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/expo/data/dataset.py b/expo/data/dataset.py
index 21dc19519..2efaf692b 100644
--- a/expo/data/dataset.py
+++ b/expo/data/dataset.py
@@ -291,6 +291,7 @@ class ExpDataset:
             train, test = train_test_split(df, test_size=1 - TRAIN_TEST_SPLIT, random_state=SEED)
         else:
             train = df
+            test = test_df
         train, dev = train_test_split(train, test_size=1 - TRAIN_DEV_SPLIT, random_state=SEED)
         self.save_split_datasets(train, "train")
         self.save_split_datasets(dev, "dev", target_col)