diff --git a/embedding_converter/src/data_loader.py b/embedding_converter/src/data_loader.py index 11f5a51..738abbc 100644 --- a/embedding_converter/src/data_loader.py +++ b/embedding_converter/src/data_loader.py @@ -6,7 +6,7 @@ import torch from torch.utils.data import Dataset from torchvision import transforms -from .types import Batch, Paths +from .types import Batch class DataLoaderRecognition(Dataset[torch.Tensor]): diff --git a/embedding_converter/src/types.py b/embedding_converter/src/types.py index a40d4bb..1a01548 100644 --- a/embedding_converter/src/types.py +++ b/embedding_converter/src/types.py @@ -3,7 +3,6 @@ from typing import Any, List, TypeAlias from numpy.typing import NDArray from torch import Tensor -Paths : TypeAlias = List[str] Batch : TypeAlias = Tensor Embedding : TypeAlias = Tensor VisionFrame : TypeAlias = NDArray[Any]