Merge pull request #408 from Renumics/bug/fix-na_mask

🐛 Fix pandas data source error if no nan values in column
Renumics · Feb 1, 2024 · 17bd7fc · 17bd7fc
2 parents d601191 + 527e0fa
commit 17bd7fc
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/renumics/spotlight_plugins/core/pandas_data_source.py b/renumics/spotlight_plugins/core/pandas_data_source.py
@@ -142,17 +142,20 @@ def get_column_values(
         if pd.api.types.is_bool_dtype(column):
             values = column.to_numpy()
             na_mask = column.isna()
-            values[na_mask] = None
+            if na_mask.any():
+                values[na_mask] = None
             return values
         if pd.api.types.is_integer_dtype(column):
             values = column.to_numpy()
             na_mask = column.isna()
-            values[na_mask] = None
+            if na_mask.any():
+                values[na_mask] = None
             return values
         if pd.api.types.is_float_dtype(column):
             values = column.to_numpy()
             na_mask = column.isna()
-            values[na_mask] = None
+            if na_mask.any():
+                values[na_mask] = None
             return values
         if pd.api.types.is_datetime64_any_dtype(column):
             return column.dt.tz_localize(None).to_numpy()