mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-26 16:27:11 +00:00
24 lines
583 B
Python
24 lines
583 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
|