Fix compile with segnext2

openvinotoolkit · Oct 7, 2024 · 2a2c52b · 2a2c52b
1 parent 54c2d49
commit 2a2c52b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/otx/algo/segmentation/backbones/mscan.py b/src/otx/algo/segmentation/backbones/mscan.py
@@ -411,7 +411,7 @@ def __init__(
         if pretrained_weights is not None:
             self.load_pretrained_weights(pretrained_weights)
 
-    def forward(self, x: torch.Tensor) -> torch.Tensor:
+    def forward(self, x: torch.Tensor) -> list[torch.Tensor]:
         """Forward function."""
         outs = []