Remove comparison to output_dir

huggingface · Apr 24, 2024 · 806dbf7 · 806dbf7
1 parent d1d94d7
commit 806dbf7
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/transformers/integrations/integration_utils.py b/src/transformers/integrations/integration_utils.py
@@ -762,9 +762,8 @@ def setup(self, args, state, model, **kwargs):
  if trial_name is not None:
  init_args["name"] = trial_name
  init_args["group"] = args.run_name
- else:
- if not (args.run_name is None or args.run_name == args.output_dir):
- init_args["name"] = args.run_name
+ elif args.run_name is not None:
+ init_args["name"] = args.run_name
 
  if self._wandb.run is None:
  self._wandb.init(