[tune] wandb log cleaning to use yaml representer (#10680)

Co-authored-by: Richard Liaw <rliaw@berkeley.edu>
This commit is contained in:
Kai Fricke
2020-09-10 08:07:16 +01:00
committed by GitHub
parent e72838c03d
commit 67bf396ae7
+8
View File
@@ -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