changing inf and nan to work with json dumps

MeltanoLabs · Oct 11, 2023 · 842a58a · 842a58a
1 parent 46d971d
commit 842a58a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tap_mongodb/streams.py b/tap_mongodb/streams.py
@@ -32,14 +32,14 @@
 
 def recursive_replace_inf_in_dict(dct):
     for key, value in dct.items():
-        if value in [-math.inf, math.inf]:
-            dct[key] = str(dct[key])
+        if value in [-math.inf, math.inf, math.nan]:
+            dct[key] = None
         elif isinstance(value, list):
             for i, item in enumerate(value):
                 if isinstance(item, dict):
                     recursive_replace_inf_in_dict(item)
-                elif item in [-math.inf, math.inf]:
-                    value[i] = str(item)
+                elif item in [-math.inf, math.inf, math.nan]:
+                    value[i] = None
         elif isinstance(value, dict):
             recursive_replace_inf_in_dict(value)
     return