From ed2130d8aaff6506510f50a5beb976dfd4e1da0c Mon Sep 17 00:00:00 2001 From: Xu Yang Date: Fri, 20 Sep 2024 14:03:46 +0800 Subject: [PATCH] fix: Update runner.py to fix a small bug (#282) * fix: Update runner.py to fix a small bug * fix CI --- rdagent/scenarios/kaggle/developer/runner.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/rdagent/scenarios/kaggle/developer/runner.py b/rdagent/scenarios/kaggle/developer/runner.py index 7fa6242a..9a7f04b0 100644 --- a/rdagent/scenarios/kaggle/developer/runner.py +++ b/rdagent/scenarios/kaggle/developer/runner.py @@ -94,7 +94,9 @@ class KGFactorRunner(KGCachedRunner[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]