fix test case when featuretools>=1.26

DataCanvasIO · May 6, 2023 · 9cbc56c · 9cbc56c
1 parent ededd1c
commit 9cbc56c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/hypernets/tests/utils/df_utils_test.py b/hypernets/tests/utils/df_utils_test.py
@@ -1,5 +1,4 @@
 from hypernets.utils import df_utils
-from hypernets.tabular.datasets import dsutils
 import numpy as np
 from sklearn.preprocessing import LabelEncoder
 

diff --git a/hypernets/utils/df_utils.py b/hypernets/utils/df_utils.py
@@ -1,8 +1,5 @@
-import json
-
 import numpy as np
 import pandas as pd
-from dask import dataframe as dd
 
 from hypernets.tabular import column_selector as col_se
 
@@ -70,6 +67,7 @@ def get_data_character(hyper_model, X_train, y_train, X_eval=None, y_eval=None,
  shape_x_test = list(X_test.shape)
 
  else:
+ from dask import dataframe as dd
  datatype_y = dtype2usagetype[str(y_train.dtype)]
 
  Missing_y = y_train.isnull().compute().tolist().count(True)
@@ -255,5 +253,5 @@ def _is_cupy_array(obj):
  elif _is_cupy_array(array_data):
  return np.array(array_data.tolist())
  else:
- logger.warning(f"unseen data type {type(array_data)} convert to numpy ndarray")
+ logger.warning(f"unseen data type {type(array_data)} to convert to ndarray")
  return array_data