From 93ff1f8f2bcf3f7304cd2282cfe17383a39532e4 Mon Sep 17 00:00:00 2001
From: Yizhou Chi <chiyizhou@fuzhi.ai>
Date: Mon, 9 Sep 2024 13:50:36 +0800
Subject: [PATCH] update datasets.yaml

---
 expo/datasets.yaml | 2 +-
 expo/utils.py      | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/expo/datasets.yaml b/expo/datasets.yaml
index 45150833a..512cbc292 100644
--- a/expo/datasets.yaml
+++ b/expo/datasets.yaml
@@ -151,7 +151,7 @@ datasets:
       \ the target column `Class`.\nPerform data analysis, data preprocessing, feature\
       \ engineering, and modeling to predict the target. \nReport f1 weighted on the\
       \ eval data. Do not plot or make any visualizations.\n"
-    concrete-strength:
+  concrete-strength:
     dataset: concrete-strength
     metric: rmse
     target_col: Strength
diff --git a/expo/utils.py b/expo/utils.py
index 65701c3ec..9c6295fa9 100644
--- a/expo/utils.py
+++ b/expo/utils.py
@@ -19,7 +19,9 @@ def load_data_config(file_path="data.yaml"):
     return data_config
 
 
+DATASET_CONFIG = load_data_config("datasets.yaml")
 DATA_CONFIG = load_data_config()
+DATA_CONFIG["datasets"].update(DATASET_CONFIG["datasets"])
 
 
 def get_mcts_logger():