From 8d8de9377929d41c1a900775acacfc7558510a96 Mon Sep 17 00:00:00 2001 From: "jiangnana.jnn" Date: Tue, 4 Apr 2023 15:01:01 +0800 Subject: [PATCH] fix json loads numpy Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/12071834 * fix jsonplus * add unittest * fix json dump np float --- modelscope/fileio/format/jsonplus.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/modelscope/fileio/format/jsonplus.py b/modelscope/fileio/format/jsonplus.py index b4da3429..8608ce93 100644 --- a/modelscope/fileio/format/jsonplus.py +++ b/modelscope/fileio/format/jsonplus.py @@ -378,6 +378,8 @@ _decode_handlers = { 'tuple': tuple, 'set': set, 'ndarray': np.asarray, + 'float16': np.float16, + 'float32': np.float32, 'frozenset': frozenset, 'complex': kwargified(complex), 'Decimal': Decimal,