change NumpyEncoder in callback

HFUT-LEC · Jul 29, 2024 · 7d723a5 · 7d723a5
1 parent 72d6a32
commit 7d723a5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 7 deletions.
diff --git a/edustudio/utils/callback/callbacks/history.py b/edustudio/utils/callback/callbacks/history.py
@@ -8,12 +8,7 @@
 import json
 import os
 import numpy as np
-
-class NumpyEncoder(json.JSONEncoder):
-    def default(self, obj):
-        if isinstance(obj, np.ndarray):
-            return obj.tolist()
-        return json.JSONEncoder.default(self, obj)
+from edustudio.utils.common import NumpyEncoder
 
 
 class History(Callback):

diff --git a/edustudio/utils/common/__init__.py b/edustudio/utils/common/__init__.py
@@ -4,5 +4,5 @@
 
 
 from .commonUtil import PathUtil, IDUtil, tensor2npy, tensor2cpu, IOUtil, set_same_seeds, get_gpu_usage, DecoratorTimer
-from .configUtil import UnifyConfig
+from .configUtil import UnifyConfig, NumpyEncoder
 from .loggerUtil import Logger