fix(NpEncoder): handle pandas NA types

OxWearables · Jun 10, 2024 · 67415b0 · 67415b0
1 parent 9712886
commit 67415b0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/stepcount/stepcount.py b/src/stepcount/stepcount.py
@@ -830,6 +830,8 @@ def default(self, obj):
             return float(obj)
         if isinstance(obj, np.ndarray):
             return obj.tolist()
+        if pd.isnull(obj):  # handles pandas NAType
+            return np.nan
         return json.JSONEncoder.default(self, obj)