More adjustments

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':
 		{

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