[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
zhiqwang · Oct 20, 2021 · 4694c37 · 4694c37
1 parent 383809f
commit 4694c37
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/yolort/utils/update_module_state.py b/yolort/utils/update_module_state.py
@@ -31,9 +31,9 @@ def load_from_ultralytics(checkpoint_path: str, version: str = "r6.0"):
     strides = checkpoint_yolov5.stride
     anchor_grids = checkpoint_yolov5.yaml["anchors"]
     if isinstance(anchor_grids, int):
-        anchor_grids = to_numpy(
-            checkpoint_yolov5.model[-1].anchor_grid
-        ).reshape(3, -1).tolist()
+        anchor_grids = (
+            to_numpy(checkpoint_yolov5.model[-1].anchor_grid).reshape(3, -1).tolist()
+        )
 
     depth_multiple = checkpoint_yolov5.yaml["depth_multiple"]
     width_multiple = checkpoint_yolov5.yaml["width_multiple"]