Refactor json_encode_np

2025-08-27 00:37:19 +00:00 · 2016-12-30 12:06:07 -08:00
parent 1059ccdf0a
commit ae8956921b
3 changed files with 22 additions and 20 deletions
--- a/gym/utils/json_utils.py
+++ b/gym/utils/json_utils.py
@@ -0,0 +1,18 @@
+import numpy as np
+
+def json_encode_np(obj):
+    """
+    JSON can't serialize numpy types, so convert to pure python
+    """
+    if isinstance(obj, np.ndarray):
+        return list(obj)
+    elif isinstance(obj, np.float32):
+        return float(obj)
+    elif isinstance(obj, np.float64):
+        return float(obj)
+    elif isinstance(obj, np.int32):
+        return int(obj)
+    elif isinstance(obj, np.int64):
+        return int(obj)
+    else:
+        return obj