diff --git a/rdagent/scenarios/kaggle/experiment/spaceship-titanic_template/fea_share_preprocess.py b/rdagent/scenarios/kaggle/experiment/spaceship-titanic_template/fea_share_preprocess.py index edf96a4a..ed889335 100644 --- a/rdagent/scenarios/kaggle/experiment/spaceship-titanic_template/fea_share_preprocess.py +++ b/rdagent/scenarios/kaggle/experiment/spaceship-titanic_template/fea_share_preprocess.py @@ -79,12 +79,12 @@ def preprocess_script(): This method applies the preprocessing steps to the training, validation, and test datasets. """ if os.path.exists("/kaggle/input/X_train.pkl"): - X_train = pd.read_pickle("/kaggle/input/X_train.pkl") - X_valid = pd.read_pickle("/kaggle/input/X_valid.pkl") - y_train = pd.read_pickle("/kaggle/input/y_train.pkl") - y_valid = pd.read_pickle("/kaggle/input/y_valid.pkl") - X_test = pd.read_pickle("/kaggle/input/X_test.pkl") - others = pd.read_pickle("/kaggle/input/others.pkl") + X_train = pd.read_pickle("/kaggle/input/X_train.pkl") # nosec B301 — trusted Kaggle input + X_valid = pd.read_pickle("/kaggle/input/X_valid.pkl") # nosec B301 + y_train = pd.read_pickle("/kaggle/input/y_train.pkl") # nosec B301 + y_valid = pd.read_pickle("/kaggle/input/y_valid.pkl") # nosec B301 + X_test = pd.read_pickle("/kaggle/input/X_test.pkl") # nosec B301 + others = pd.read_pickle("/kaggle/input/others.pkl") # nosec B301 y_train = pd.Series(y_train).reset_index(drop=True) y_valid = pd.Series(y_valid).reset_index(drop=True)