gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-07-29 08:27:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix minor bug in feature test (#531)

Browse Source
This commit is contained in:
Xu Yang
2025-01-23 10:15:21 +08:00
committed by GitHub
parent 7ff55f0013
commit f19a91abfc
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/components/coder/data_science/feature/eval_tests/feature_test.txt
+7 -4
View File
@@ -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."
)