fix(pre-commit.ci): auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
kraina-ai · Aug 5, 2024 · 64e1daa · 64e1daa
1 parent 9015e20
commit 64e1daa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/srai/embedders/gtfs2vec/model.py b/srai/embedders/gtfs2vec/model.py
@@ -56,7 +56,7 @@ def forward(self, x: "torch.Tensor") -> "torch.Tensor":
         Args:
             x (torch.Tensor): Input tensor.
         """
-        embedding: "torch.Tensor" = self.encoder(x)
+        embedding: torch.Tensor = self.encoder(x)
         return embedding
 
     def configure_optimizers(self) -> "torch.optim.Optimizer":

diff --git a/srai/embedders/highway2vec/model.py b/srai/embedders/highway2vec/model.py
@@ -60,7 +60,7 @@ def forward(self, x: "torch.Tensor") -> "torch.Tensor":
         Args:
             x (torch.Tensor): Input tensor.
         """
-        z: "torch.Tensor" = self.encoder(x)
+        z: torch.Tensor = self.encoder(x)
         return z
 
     def training_step(self, batch: "torch.Tensor", batch_idx: int) -> "torch.Tensor":