[tune] Remove unnecessary wandb group parameter (#10950)

This commit is contained in:
Kai Fricke
2020-09-22 17:36:51 +01:00
committed by Barak Michener
parent f371eb334c
commit 13d703be12
+4 -4
View File
@@ -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)