mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-27 16:57:10 +00:00
23 lines
582 B
Python
23 lines
582 B
Python
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.int8):
|
|
return int(obj)
|
|
elif isinstance(obj, np.int16):
|
|
return int(obj)
|
|
elif isinstance(obj, np.int32):
|
|
return int(obj)
|
|
elif isinstance(obj, np.int64):
|
|
return int(obj)
|
|
else:
|
|
return obj
|