lint (#81)

facebookresearch · Oct 11, 2023 · 6944d63 · 6944d63
1 parent 5aabaf1
commit 6944d63
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/simuleval/agents/pipeline.py b/simuleval/agents/pipeline.py
@@ -303,7 +303,7 @@ def push(
             states = {module: None for module in self.module_dict}
         else:
             assert len(states) == len(self.module_dict)
-    
+
         if upstream_states is None:
             upstream_states = {}
 

diff --git a/simuleval/options.py b/simuleval/options.py
@@ -178,9 +178,7 @@ def general_parser():
     parser.add_argument(
         "--device", type=str, default="cpu", help="Device to run the model."
     )
-    parser.add_argument(
-        "--fp16", action="store_true", default=False, help="Use fp16."
-    )
+    parser.add_argument("--fp16", action="store_true", default=False, help="Use fp16.")
     return parser