From 7ab7efbbf4af3a26d9b31b08a73a21c11e96b879 Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Thu, 6 Mar 2025 13:18:52 +0100
Subject: [PATCH] More adjustments

---
 embedding_converter/src/training.py | 4 ++--
 embedding_converter/src/types.py    | 5 +++--
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/embedding_converter/src/training.py b/embedding_converter/src/training.py
index 02d093a..a492df9 100644
--- a/embedding_converter/src/training.py
+++ b/embedding_converter/src/training.py
@@ -12,7 +12,7 @@ from torchdata.stateful_dataloader import StatefulDataLoader
 
 from .dataset import StaticDataset
 from .models.embedding_converter import EmbeddingConverter
-from .types import Batch, Config, Embedding, OptimizerSet
+from .types import Batch, Config, ConfigSet, Embedding, OptimizerSet
 
 CONFIG = configparser.ConfigParser()
 CONFIG.read('config.ini')
@@ -121,7 +121,7 @@ def create_trainer() -> Trainer:
 
 
 def train() -> None:
-	config : Config =\
+	config : ConfigSet =\
 	{
 		'dataset':
 		{
diff --git a/embedding_converter/src/types.py b/embedding_converter/src/types.py
index fd6235b..38a5708 100644
--- a/embedding_converter/src/types.py
+++ b/embedding_converter/src/types.py
@@ -1,9 +1,10 @@
-from typing import Any, TypeAlias
+from typing import Any, Dict, TypeAlias
 
 from torch import Tensor
 
 Batch : TypeAlias = Tensor
 Embedding : TypeAlias = Tensor
 
-Config : TypeAlias = Any
+Config : TypeAlias = Dict[str, Any]
+ConfigSet : TypeAlias = Dict[str, Config]
 OptimizerSet : TypeAlias = Any