From 40786810310cb49cd04cd59f17c29618860fd7f7 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Fri, 21 Feb 2025 09:17:59 +0100 Subject: [PATCH] Fix typing --- embedding_converter/src/training.py | 2 +- embedding_converter/src/types.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/embedding_converter/src/training.py b/embedding_converter/src/training.py index ab31c2b..64be995 100644 --- a/embedding_converter/src/training.py +++ b/embedding_converter/src/training.py @@ -83,7 +83,7 @@ def create_loaders(dataset : Dataset[Tensor]) -> Tuple[DataLoader[Tensor], DataL def split_dataset(dataset : Dataset[Tensor]) -> Tuple[Dataset[Tensor], Dataset[Tensor]]: loader_split_ratio = CONFIG.getfloat('training.loader', 'split_ratio') - dataset_size = len(dataset) + dataset_size = len(dataset) # type:ignore[arg-type] training_size = dataset_size * loader_split_ratio validation_size = dataset_size - training_size training_dataset, validate_dataset = random_split(dataset, [ training_size, validation_size ]) diff --git a/embedding_converter/src/types.py b/embedding_converter/src/types.py index 62db931..7513a43 100644 --- a/embedding_converter/src/types.py +++ b/embedding_converter/src/types.py @@ -7,4 +7,4 @@ Batch : TypeAlias = Tensor Embedding : TypeAlias = Tensor VisionFrame : TypeAlias = NDArray[Any] -OptimizerConfig : TypeAlias = Dict[str, Any] +OptimizerConfig : TypeAlias = Any