Make device parameter work like DLMUSE and DLICV

CBICA · Jan 10, 2025 · 5d746bb · 5d746bb
1 parent 2418928
commit 5d746bb
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/NiChart_DLMUSE/__main__.py b/NiChart_DLMUSE/__main__.py
@@ -64,8 +64,7 @@ def main() -> None:
 
     # DEVICE argument
     parser.add_argument(
-        "-d",
-        "--device",
+        "-device",
         type=str,
         help="Device (cpu, cuda, or mps)",
         default=None,