21667: Fixes a bug with multiprocesing IFA (#296)

howsoai · Sep 25, 2024 · d0b5109 · d0b5109
1 parent e7201d9
commit d0b5109
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/howso/utilities/feature_attributes/pandas.py b/howso/utilities/feature_attributes/pandas.py
@@ -44,7 +44,7 @@ def _shard(data: pd.DataFrame, *, kwargs: dict[str, t.Any]):
     ifr_inst = InferFeatureAttributesDataFrame(data)
     # Filter out features that are not related to this shard.
     _kwargs = kwargs.copy()
-    if "features" in _kwargs:
+    if _kwargs.get("features") is not None:
         _kwargs['features'] = {
             k: v for k, v in _kwargs["features"].items()
             if k in data.columns