fix: linting

chanind · Aug 26, 2023 · b331dfa · b331dfa
1 parent 6f2e1d1
commit b331dfa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/frame_semantic_transformer/scripts/train.py b/frame_semantic_transformer/scripts/train.py
@@ -22,13 +22,13 @@
     parser.add_argument(
         "--use-gpu",
         default=torch.cuda.is_available(),
-        action='store_true',
+        action="store_true",
         help=f"Whether to use GPU for training, default {torch.cuda.is_available()}",
     )
     parser.add_argument(
         "--use-cpu",
         dest="use_gpu",
-        action='store_false',
+        action="store_false",
         help=f"Whether to use CPU for training, default {not torch.cuda.is_available()}",
     )
     parser.add_argument("--learning-rate", default=1e-4, type=float)