From 38daf24c33f9e89dea3fa53e772924cd0daae16c Mon Sep 17 00:00:00 2001
From: Yizhou Chi <chiyizhou@fuzhi.ai>
Date: Thu, 17 Oct 2024 09:55:37 +0800
Subject: [PATCH] rename task if custom_data_dir is used

---
 expo/run_experiment.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/expo/run_experiment.py b/expo/run_experiment.py
index 4e6b41fd7..c977b4dc9 100644
--- a/expo/run_experiment.py
+++ b/expo/run_experiment.py
@@ -1,6 +1,7 @@
 import argparse
 import asyncio
 
+from expo.data.custom_task import get_mle_task_id
 from expo.experimenter.aug import AugExperimenter
 from expo.experimenter.autogluon import GluonExperimenter
 from expo.experimenter.autosklearn import AutoSklearnExperimenter
@@ -68,6 +69,7 @@ async def main(args):
         args.external_eval = False
         args.eval_func = "mlebench"
         args.from_scratch = True
+        args.task = get_mle_task_id(args.custom_dataset_dir)
 
     if args.exp_mode == "mcts":
         experimenter = MCTSExperimenter(args)