mirror of
https://github.com/modelscope/modelscope.git
synced 2026-02-24 20:19:51 +01:00
dbacead74e
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/9703595 * fix json dump numpy * Merge remote-tracking branch 'origin' into fix/josn_dump
18 lines
478 B
Python
18 lines
478 B
Python
# Copyright (c) Alibaba, Inc. and its affiliates.
|
|
|
|
import json
|
|
import numpy as np
|
|
|
|
|
|
class EnhancedEncoder(json.JSONEncoder):
|
|
""" Enhanced json encoder for not supported types """
|
|
|
|
def default(self, obj):
|
|
if isinstance(obj, np.integer):
|
|
return int(obj)
|
|
elif isinstance(obj, np.floating):
|
|
return float(obj)
|
|
elif isinstance(obj, np.ndarray):
|
|
return obj.tolist()
|
|
return json.JSONEncoder.default(self, obj)
|