fix: Update runner.py to fix a small bug (microsoft#282)

* fix: Update runner.py to fix a small bug * fix CI
shreyanshp · Sep 20, 2024 · 8aef3ab · 8aef3ab
1 parent 4c8f998
commit 8aef3ab
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/rdagent/scenarios/kaggle/developer/runner.py b/rdagent/scenarios/kaggle/developer/runner.py
@@ -94,7 +94,9 @@ def init_develop(self, exp: KGFactorExperiment) -> KGFactorExperiment:
             factor_name="original features", factor_description="here is the original features", factor_formulation=""
         )
 
-        org_data_path = Path(FACTOR_IMPLEMENT_SETTINGS.data_folder) / KAGGLE_IMPLEMENT_SETTING.competition / "valid.pkl"
+        org_data_path = (
+            Path(FACTOR_IMPLEMENT_SETTINGS.data_folder) / KAGGLE_IMPLEMENT_SETTING.competition / "X_valid.pkl"
+        )
         with open(org_data_path, "rb") as f:
             org_data = pickle.load(f)
         feature_shape = org_data.shape[-1]