swaglog: support np.bool_ (#34439)

fix np.bool_
pull/34440/head
Shane Smiskol 3 months ago committed by GitHub
parent ffa5834225
commit 9ceb8cc324
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 3
      common/logging_extra.py

@ -8,6 +8,7 @@ import uuid
import socket import socket
import logging import logging
import traceback import traceback
import numpy as np
from threading import local from threading import local
from collections import OrderedDict from collections import OrderedDict
from contextlib import contextmanager from contextlib import contextmanager
@ -15,6 +16,8 @@ from contextlib import contextmanager
LOG_TIMESTAMPS = "LOG_TIMESTAMPS" in os.environ LOG_TIMESTAMPS = "LOG_TIMESTAMPS" in os.environ
def json_handler(obj): def json_handler(obj):
if isinstance(obj, np.bool_):
return bool(obj)
# if isinstance(obj, (datetime.date, datetime.time)): # if isinstance(obj, (datetime.date, datetime.time)):
# return obj.isoformat() # return obj.isoformat()
return repr(obj) return repr(obj)

Loading…
Cancel
Save