From 5c56760faca2c088101b6a583155ec0584240c8a Mon Sep 17 00:00:00 2001 From: Eli Meirom Date: Sat, 13 Jun 2020 02:39:01 +0300 Subject: [PATCH] [tune] np.array compat for logger (#8918) Co-authored-by: Richard Liaw --- python/ray/tune/logger.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python/ray/tune/logger.py b/python/ray/tune/logger.py index a996e2d0f..044448d47 100644 --- a/python/ray/tune/logger.py +++ b/python/ray/tune/logger.py @@ -218,7 +218,9 @@ class TBXLogger(Logger): valid_result[full_attr] = value self._file_writer.add_scalar( full_attr, value, global_step=step) - elif type(value) in [list, np.ndarray] and len(value) > 0: + elif (type(value) == list + and len(value) > 0) or (type(value) == np.ndarray + and value.size > 0): valid_result[full_attr] = value try: self._file_writer.add_histogram(