chore: lint

axolotl-ai-cloud · Apr 9, 2024 · f819da7 · f819da7
1 parent 6f8ed88
commit f819da7
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/axolotl/utils/callbacks/__init__.py b/src/axolotl/utils/callbacks/__init__.py
@@ -748,9 +748,11 @@ def on_train_begin(
                     mode="w", delete=False, suffix=".yml", prefix="axolotl_config_"
                 ) as temp_file:
                     copyfile(self.axolotl_config_path, temp_file.name)
-                    at = wandb.Artifact(f"config-{wandb.run.id}", type="axolotl-config")
-                    at.add_file(temp_file.name)
-                    wandb.log_artifact(at)
+                    artifact = wandb.Artifact(
+                        f"config-{wandb.run.id}", type="axolotl-config"
+                    )
+                    artifact.add_file(temp_file.name)
+                    wandb.log_artifact(artifact)
                     wandb.save(temp_file.name)
                 LOG.info(
                     "The Axolotl config has been saved to the WandB run under files."