add typing

gau-nernst · Jul 24, 2023 · ad63099 · ad63099
1 parent ff4eba2
commit ad63099
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vision_toolbox/backbones/mobile_vit.py b/vision_toolbox/backbones/mobile_vit.py
@@ -92,7 +92,7 @@ class MobileViTStageConfig(NamedTuple):
 
 
 class MobileViT(BaseBackbone):
-    def __init__(self, stem_channels: int, stage_configs: list[MobileViTStageConfig], out_channels: int):
+    def __init__(self, stem_channels: int, stage_configs: list[MobileViTStageConfig], out_channels: int) -> None:
         super().__init__()
         self.stem = nn.Sequential(
             ConvNormAct(3, 16, 3, 2, act="swish"),