style: Update variable name in predict function

ylab-hi · Oct 13, 2024 · 9b73043 · 9b73043
1 parent 5fc1d3c
commit 9b73043
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -593,4 +593,5 @@ scripts/*.json
 *stats_*.json
 !build.rs
 .ruff_cache
-hg_deepchopper
+hg_deepchopper
+analysis_data
diff --git a/deepchopper/cli.py b/deepchopper/cli.py
@@ -135,7 +135,7 @@ def predict(
 
     trainer = lightning.pytorch.trainer.Trainer(
         accelerator=accelerator,
-        devices=gpus,
+        devices=devices,
         callbacks=callbacks,
         deterministic=False,
         logger=False,