ray packages moved around :(
import numbers
import wandb
from ray import tune
+
+# ray 0.8.1 reorganized ray.tune.util -> ray.tune.utils
+try:
+ from ray.tune.utils import flatten_dict
+except ImportError:
from ray.tune.util import flatten_dict
