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

ignore new numeric type (#638)

Browse Source
This commit is contained in:
Tim
2025-02-25 18:20:14 +08:00
committed by GitHub
parent c109fa8e99
commit 36dfd6d424
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/components/coder/data_science/feature/eval_tests/feature_test.txt
+5 -2
View File
@@ -61,8 +61,11 @@ 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):
X_dtypes_unique_sorted = sorted([str(dt) for dt in X.dtypes.unique()])
X_loaded_dtypes_unique_sorted = sorted([str(dt) for dt in X_loaded.dtypes.unique()])
def normalize_dtype(dtype):
return "numeric" if np.issubdtype(dtype, np.number) else str(dtype)
X_dtypes_unique_sorted = sorted(set(normalize_dtype(dt) for dt in X.dtypes.unique()))
X_loaded_dtypes_unique_sorted = sorted(set(normalize_dtype(dt) for dt in X_loaded.dtypes.unique()))
X_dtypes_unique_sorted_new = [
dt for dt in X_dtypes_unique_sorted if dt not in X_loaded_dtypes_unique_sorted and dt != "object"