fix lint

open-mmlab · Dec 13, 2023 · 97231f8 · 97231f8
1 parent 2207861
commit 97231f8
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/mmdeploy/codebase/base/task.py b/mmdeploy/codebase/base/task.py
@@ -127,9 +127,10 @@ def build_pytorch_model(self,
                                                   'switch_to_deploy'):
             model.backbone.switch_to_deploy()
 
-        if hasattr(model, 'switch_to_deploy') and callable(model.switch_to_deploy):
-            model.switch_to_deploy() 
-
+        if hasattr(model, 'switch_to_deploy') and callable(
+                model.switch_to_deploy):
+            model.switch_to_deploy()
+
         model = model.to(self.device)
         model.eval()
         return model