make featuretools optional

DataCanvasIO · May 8, 2023 · de6cf28 · de6cf28
1 parent a4a9aeb
commit de6cf28
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hypernets/tabular/dask_ex/_toolbox.py b/hypernets/tabular/dask_ex/_toolbox.py
@@ -825,7 +825,7 @@ def compute_sample_weight(y):
 )
 
 if _feature_generators.is_feature_generator_ready:
- _predefined_transformers['FeatureGenerationTransformer'] = _feature_generators.FeatureGenerationTransformer
+ _predefined_transformers['FeatureGenerationTransformer'] = _feature_generators.DaskFeatureGenerationTransformer
 
 for name, tf in _predefined_transformers.items():
  register_transformer(tf, name=name, dtypes=dd.DataFrame)