From 6346ea0be6b977c5259843c08ffe7b39f698dfc0 Mon Sep 17 00:00:00 2001
From: Sumanth Ratna <sumanthratna@gmail.com>
Date: Mon, 23 Nov 2020 15:57:23 -0500
Subject: [PATCH] [tune] Allow None values in TBX logger (#12262)

---
 python/ray/tune/logger.py            | 4 ++--
 python/ray/tune/tests/test_logger.py | 3 ++-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/python/ray/tune/logger.py b/python/ray/tune/logger.py
index ff55676cc..16765d09e 100644
--- a/python/ray/tune/logger.py
+++ b/python/ray/tune/logger.py
@@ -199,8 +199,8 @@ class TBXLogger(Logger):
         {"a": {"b": 1, "c": 2}} -> {"a/b": 1, "a/c": 2}
     """
 
-    # NoneType is not supported on the last TBX release yet.
-    VALID_HPARAMS = (str, bool, np.bool8, int, np.integer, float, list)
+    VALID_HPARAMS = (str, bool, np.bool8, int, np.integer, float, list,
+                     type(None))
 
     def _init(self):
         try:
diff --git a/python/ray/tune/tests/test_logger.py b/python/ray/tune/tests/test_logger.py
index 234b65c43..e60e681df 100644
--- a/python/ray/tune/tests/test_logger.py
+++ b/python/ray/tune/tests/test_logger.py
@@ -147,7 +147,8 @@ class LoggerSuite(unittest.TestCase):
                 }
             },
             "d": np.int64(1),
-            "e": np.bool8(True)
+            "e": np.bool8(True),
+            "f": None,
         }
         t = Trial(
             evaluated_params=config, trial_id="tbx", logdir=self.test_dir)