fix :Object of type int32 is not JSON serializable

0c5de3ac · Zhiwei Zhuang · a007bacb · 0c5de3ac
Commit 0c5de3ac authored Oct 19, 2023 by Zhiwei Zhuang
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

lm_eval/__main__.py lm_eval/__main__.py +1 -1

No files found.
--- a/lm_eval/__main__.py
+++ b/lm_eval/__main__.py
@@ -15,7 +15,7 @@ from typing import Union


 def _handle_non_serializable(o):
-    if isinstance(o, np.int64):
+    if isinstance(o, np.int64) or isinstance(o, np.int32):
        return int(o)
    elif isinstance(o, set):
        return list(o)