From 67bf396ae7b130a0df15d6adffe45da78c15524f Mon Sep 17 00:00:00 2001
From: Kai Fricke <krfricke@users.noreply.github.com>
Date: Thu, 10 Sep 2020 08:07:16 +0100
Subject: [PATCH] [tune] wandb log cleaning to use yaml representer (#10680)

Co-authored-by: Richard Liaw <rliaw@berkeley.edu>
---
 python/ray/tune/integration/wandb.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/python/ray/tune/integration/wandb.py b/python/ray/tune/integration/wandb.py
index b464187a2..b3248cbcb 100644
--- a/python/ray/tune/integration/wandb.py
+++ b/python/ray/tune/integration/wandb.py
@@ -9,6 +9,8 @@ from ray.tune.function_runner import FunctionRunner
 from ray.tune.logger import Logger
 from ray.tune.utils import flatten_dict
 
+import yaml
+
 try:
     import wandb
 except ImportError:
@@ -29,6 +31,12 @@ def _clean_log(obj):
     # Else
     try:
         pickle.dumps(obj)
+        yaml.dump(
+            obj,
+            Dumper=yaml.SafeDumper,
+            default_flow_style=False,
+            allow_unicode=True,
+            encoding="utf-8")
         return obj
     except Exception:
         # give up, similar to _SafeFallBackEncoder