From 9ceb8cc3243cc16855d8c03b10f92642db30a9e6 Mon Sep 17 00:00:00 2001
From: Shane Smiskol <shane@smiskol.com>
Date: Tue, 21 Jan 2025 12:52:51 -0800
Subject: [PATCH] swaglog: support np.bool_ (#34439)

fix np.bool_
---
 common/logging_extra.py | 3 +++
 1 file changed, 3 insertions(+)

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