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