From 6944d6313fac39e2d3f6ca1a0384f43e4aae1b11 Mon Sep 17 00:00:00 2001 From: Anna Sun <13106449+annasun28@users.noreply.github.com> Date: Wed, 11 Oct 2023 15:10:00 -0700 Subject: [PATCH] lint (#81) --- simuleval/agents/pipeline.py | 2 +- simuleval/options.py | 4 +--- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/simuleval/agents/pipeline.py b/simuleval/agents/pipeline.py index 5489729e..1631c3f8 100644 --- 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 index 8eced511..de9f4f75 100644 --- 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