One more type:ignore

wearepal · Feb 24, 2024 · 831304a · 831304a
1 parent 3ae31f0
commit 831304a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/labelling/encoder.py b/src/labelling/encoder.py
@@ -39,13 +39,13 @@ def __init__(
         model, self.transforms = clip.load(
             name=version.value,  # type: ignore
             device="cpu",
-            download_root=download_root,
+            download_root=download_root,  # type: ignore
         )
         logger.info("Done.")
         self.encoder = model.visual
         self.out_dim = cast(int, self.encoder.output_dim)
 
-    def forward(self, x: Tensor) -> Tensor:  # type: ignore
+    def forward(self, x: Tensor) -> Tensor:
         return self.encoder(x)
 
     @torch.no_grad()  # pyright: ignore