From 4337701bc904660afd0b35956c4597f0ccf84cb6 Mon Sep 17 00:00:00 2001 From: Pavel Iakubovskii Date: Wed, 24 Apr 2024 18:05:54 +0000 Subject: [PATCH] Remove comparison to output_dir --- src/transformers/integrations/integration_utils.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/transformers/integrations/integration_utils.py b/src/transformers/integrations/integration_utils.py index 45507bfda82198..228c2b22cad606 100644 --- 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(