From f3409c5aded789f12da9a3f7dabee9a00e712a1b Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Wed, 12 Feb 2025 17:08:12 +0100
Subject: [PATCH] Modernize to use ModuleList, Fix some types

---
 embedding_converter/src/models/embedding_converter.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/embedding_converter/src/models/embedding_converter.py b/embedding_converter/src/models/embedding_converter.py
index d66ba9a..5490aa3 100644
--- a/embedding_converter/src/models/embedding_converter.py
+++ b/embedding_converter/src/models/embedding_converter.py
@@ -1,7 +1,7 @@
 import torch
 import torch.nn as nn
 
-from embedding_converter.src.types import VisionTensor
+from ..types import VisionTensor
 
 
 class EmbeddingConverter(nn.Module):
@@ -21,7 +21,7 @@ class EmbeddingConverter(nn.Module):
 		])
 		return layers
 
-	def forward(self, input_tensor: VisionTensor) -> VisionTensor:
+	def forward(self, input_tensor : VisionTensor) -> VisionTensor:
 		output_tensor = input_tensor / torch.norm(input_tensor)
 
 		for layer in self.layers[:-1]: