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

fix: refine prompt to generate the most simple task in init stage (#546)

Browse Source 
* refine prompt to generate the most simple task in init stage

* feature test dtype check improve
This commit is contained in:
Xu Yang
2025-01-27 00:51:40 +08:00
committed by GitHub
parent b05cd1a1a7
commit 1bd192e061
2 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/components/coder/data_science/feature/eval_tests/feature_test.txt
+8 -3
View File
@@ -57,9 +57,14 @@ 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."
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()}"
X_dtypes_unique_sorted = sorted(X.dtypes.unique().tolist())
X_loaded_dtypes_unique_sorted = sorted(X_loaded.dtypes.unique().tolist())
assert (
len(X_loaded_dtypes_unique_sorted) == 1
and (X_loaded_dtypes_unique_sorted[0] == np.float64 or X_loaded_dtypes_unique_sorted[0] == np.float32)
) or (
X_dtypes_unique_sorted == X_loaded_dtypes_unique_sorted
), f"feature engineering has produced new data types which is not allowed, data loader data types are {X_loaded_dtypes_unique_sorted} and feature engineering data types are {X_dtypes_unique_sorted}"
print(
"Feature Engineering test passed successfully. All checks including length, width, and data types have been validated."