From 64e1daaef3b25ab9eb1b52dc000939c89b4eaee7 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Mon, 5 Aug 2024 23:25:37 +0000
Subject: [PATCH] fix(pre-commit.ci): auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 srai/embedders/gtfs2vec/model.py    | 2 +-
 srai/embedders/highway2vec/model.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/srai/embedders/gtfs2vec/model.py b/srai/embedders/gtfs2vec/model.py
index 69077842..18a189a8 100644
--- 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
index 4f233590..fb66bba3 100644
--- 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":