From 13d703be12f61db391ab3f4ba468376ee169e9ce Mon Sep 17 00:00:00 2001 From: Kai Fricke Date: Tue, 22 Sep 2020 17:36:51 +0100 Subject: [PATCH] [tune] Remove unnecessary wandb group parameter (#10950) --- python/ray/tune/integration/wandb.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/python/ray/tune/integration/wandb.py b/python/ray/tune/integration/wandb.py index 7f8b35c14..f62e3264a 100644 --- a/python/ray/tune/integration/wandb.py +++ b/python/ray/tune/integration/wandb.py @@ -138,7 +138,6 @@ def wandb_mixin(func): """ func.__mixins__ = (WandbTrainableMixin, ) - func.__wandb_group__ = func.__name__ return func @@ -342,8 +341,8 @@ class WandbLogger(Logger): exclude_results += ["config"] # Fill trial ID and name - trial_id = self.trial.trial_id - trial_name = str(self.trial) + trial_id = self.trial.trial_id if self.trial else None + trial_name = str(self.trial) if self.trial else None # Project name for Wandb try: @@ -353,7 +352,8 @@ class WandbLogger(Logger): "You need to specify a `project` in your wandb `config` dict.") # Grouping - wandb_group = wandb_config.pop("group", self.trial.trainable_name) + wandb_group = wandb_config.pop( + "group", self.trial.trainable_name if self.trial else None) # remove unpickleable items! config = _clean_log(config)