Fix codejail numpy encoder for

korbit-ai · Mar 18, 2022 · 1d437b4 · 1d437b4
1 parent 954acf0
commit 1d437b4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/codejail/custom_encoder.py b/codejail/custom_encoder.py
@@ -8,6 +8,7 @@ def default(self, obj):
         elif type(obj).__module__ == 'pandas':
             if hasattr(obj, 'to_json'):
                 return obj.to_json(orient='records')
+            return repr(obj)
         return json.JSONEncoder.default(self, obj)
 
 
@@ -28,6 +29,5 @@ def default(self, obj):
             return obj.tolist()
         elif isinstance(obj, (np.bool_)):
             return bool(obj)
-        elif isinstance(obj, (np.void)):
-            return None
-        return json.JSONEncoder.default(self, obj)
+        return json.JSONEncoder.default(self, repr(obj))
+