Merge branch 'main' into 21855-hyperparam-map

howsoai · Oct 22, 2024 · 7f2d330 · 7f2d330
2 parents d3c5c5c + 91cc2a6
commit 7f2d330
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/howso/utilities/feature_attributes/pandas.py b/howso/utilities/feature_attributes/pandas.py
@@ -452,7 +452,9 @@ def _as_float(value, dtype):
                 min_value, max_value = self._get_min_max_number_size_bounds(
                     feature_attributes, feature_name)
                 if min_value is not None and max_value is not None:
-                    output = {'min': min_value, 'max': max_value}
+                    output = {'min': min_value, 'max': max_value, 'allow_null': allow_null}
+                else:
+                    output = {'allow_null': allow_null}
 
         else:  # Ordinals
             output = {'allow_null': allow_null}

diff --git a/howso/utilities/feature_attributes/tests/test_infer_feature_attributes.py b/howso/utilities/feature_attributes/tests/test_infer_feature_attributes.py
@@ -339,6 +339,7 @@ def test_dependent_features(should_include, base_features, dependent_features):
     (None, [2, 2, 2, 2, 4, 5, 6, 6, 6, 6], {'min': 2, 'max': 6, 'allow_null': False}),
     (None, [2, 2, 2, 2, 4, 5, 6, 6, 6, 6, 6], {'min': 1, 'max': 6, 'allow_null': False}),
     (None, [2, 2, 2, 2, 4, 5, 6, 7], {'min': 2, 'max': 7, 'allow_null': False}),
+    (None, [float('nan'), float('nan')], {'allow_null': True}),
     (['a'], [2, 3, 4, 5, 6, 7], {'min': 2, 'max': 7, 'allow_null': False}),
     (['a'], [2, 3, 4, None, 6, 7], {'min': 2, 'max': 7, 'allow_null': True}),
     (