... fix again

awi-response · Dec 17, 2024 · 338ee90 · 338ee90
1 parent db5c319
commit 338ee90
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/darts/src/darts/training.py b/darts/src/darts/training.py
@@ -505,7 +505,7 @@ def run():
             plot_every_n_val_epochs=plot_every_n_val_epochs,
             # Device and Manager config
             num_workers=num_workers,
-            devices=device,
+            device=device,
             wandb_entity=wandb_entity,
             wandb_project=wandb_project,
             run_name=wandb.run.name,