Skip to content

GitLab

Commit 0c5de3ac authored by Zhiwei Zhuang's avatar Zhiwei Zhuang
Browse files

fix :Object of type int32 is not JSON serializable

parent a007bacb
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
lm_eval/__main__.py
View file @ 0c5de3ac
...@@ -15,7 +15,7 @@ from typing import Union ...@@ -15,7 +15,7 @@ from typing import Union
def _handle_non_serializable(o): def _handle_non_serializable(o):
if isinstance(o, np.int64): if isinstance(o, np.int64) or isinstance(o, np.int32):
return int(o) return int(o)
elif isinstance(o, set): elif isinstance(o, set):
return list(o) return list(o)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment