fix a bug in factor_execution_template_v2

microsoft · Nov 25, 2024 · a6a097d · a6a097d
1 parent 2d588b7
commit a6a097d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/rdagent/components/coder/factor_coder/factor_execution_template_v2.txt b/rdagent/components/coder/factor_coder/factor_execution_template_v2.txt
@@ -1,5 +1,5 @@
 import os
-import h5py
+
 import numpy as np
 import pandas as pd
 from feat01 import feat_eng
@@ -12,5 +12,5 @@ else:
 
 X, y, p = feat_eng(X, y)
 
-with h5py.File('result.h5', 'w') as hf:
-    hf.create_dataset('default', data=X)
+X = pd.dataframe(X)
+X.to_hdf("result.h5", key="data", mode="w")
diff --git a/rdagent/scenarios/kaggle/experiment/scenario.py b/rdagent/scenarios/kaggle/experiment/scenario.py
@@ -128,9 +128,7 @@ def background(self) -> str:
             Path(__file__).resolve()
             / Path(KAGGLE_IMPLEMENT_SETTING.template_path).resolve()
             / KAGGLE_IMPLEMENT_SETTING.competition
-            / "train.py"
-            if KAGGLE_IMPLEMENT_SETTING.template_path != "rdagent/scenarios/kaggle/tpl_ex"
-            else "main.py"
+            / ("train.py" if KAGGLE_IMPLEMENT_SETTING.template_path != "rdagent/scenarios/kaggle/tpl_ex" else "main.py")
         ).read_text()
 
         background_prompt = (