From f19a91abfcc3ec6c66bc214976c70d828067f069 Mon Sep 17 00:00:00 2001 From: Xu Yang Date: Thu, 23 Jan 2025 10:15:21 +0800 Subject: [PATCH] fix minor bug in feature test (#531) --- .../data_science/feature/eval_tests/feature_test.txt | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/rdagent/components/coder/data_science/feature/eval_tests/feature_test.txt b/rdagent/components/coder/data_science/feature/eval_tests/feature_test.txt index e0e487e6..15d8055c 100644 --- a/rdagent/components/coder/data_science/feature/eval_tests/feature_test.txt +++ b/rdagent/components/coder/data_science/feature/eval_tests/feature_test.txt @@ -56,8 +56,11 @@ assert get_width(X) == get_width( if isinstance(X, pd.DataFrame) and isinstance(X_test, pd.DataFrame): assert get_column_list(X) == get_column_list(X_test), "Mismatch in column names of training and test data." -assert sorted(X.dtypes.unique().tolist()) == sorted( - X_loaded.dtypes.unique().tolist() -), f"feature engineering has produced new data types which is not allowed, data loader data types are {X_loaded.dtypes.unique().tolist()} and feature engineering data types are {X.dtypes.unique().tolist()}" +if isinstance(X, pd.DataFrame): + assert sorted(X.dtypes.unique().tolist()) == sorted( + X_loaded.dtypes.unique().tolist() + ), f"feature engineering has produced new data types which is not allowed, data loader data types are {X_loaded.dtypes.unique().tolist()} and feature engineering data types are {X.dtypes.unique().tolist()}" -print("Feature Engineering test passed successfully. All checks including length, width, and data types have been validated.") +print( + "Feature Engineering test passed successfully. All checks including length, width, and data types have been validated." +)