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

for more information, see https://pre-commit.ci
zhiqwang · Oct 8, 2021 · 2b52e89 · 2b52e89
1 parent c08d405
commit 2b52e89
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/yolort/models/darknet.py b/yolort/models/darknet.py
@@ -6,14 +6,14 @@
     darknet_s_r4_0,
     darknet_m_r4_0,
     darknet_l_r4_0,
-    __all__ as dv5_all
+    __all__ as dv5_all,
 )
 from .darknetv6 import (
     DarkNetV6,
     darknet_s_r6_0,
     darknet_m_r6_0,
     darknet_l_r6_0,
-    __all__ as dv6_all
+    __all__ as dv6_all,
 )
 
 __all__ = dv5_all + dv6_all
diff --git a/yolort/models/darknetv5.py b/yolort/models/darknetv5.py
@@ -43,6 +43,7 @@ class DarkNet(nn.Module):
             a multiple of this number. Set to 1 to turn off rounding
         num_classes (int): Number of classes
     """
+
     def __init__(
         self,
         depth_multiple: float,