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

This commit is contained in:
Kai Fricke
2020-09-22 17:36:51 +01:00
committed by GitHub
parent 770c3633f0
commit 2d16ab2e16
+4 -4
View File
@@ -139,7 +139,6 @@ def wandb_mixin(func: Callable):
"""
func.__mixins__ = (WandbTrainableMixin, )
func.__wandb_group__ = func.__name__
return func
@@ -344,8 +343,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:
@@ -355,7 +354,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)