From e33bc0d52a272bee19ae3fb6b29bde23da6f7081 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Wed, 12 Feb 2025 17:15:15 +0100 Subject: [PATCH] Modernize to use ModuleList, Fix some types --- embedding_converter/src/types.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/embedding_converter/src/types.py b/embedding_converter/src/types.py index 8b18e85..9221412 100644 --- a/embedding_converter/src/types.py +++ b/embedding_converter/src/types.py @@ -1,13 +1,13 @@ -from typing import Any, Tuple +from typing import Any, Tuple, TypeAlias from numpy.typing import NDArray from torch import Tensor -Embedding = NDArray[Any] -EmbeddingDataset = NDArray[Embedding] -FaceLandmark5 = NDArray[Any] +Embedding : TypeAlias = NDArray[Any] +EmbeddingDataset : TypeAlias = NDArray[Embedding] +FaceLandmark5 : TypeAlias = NDArray[Any] -VisionFrame = NDArray[Any] -VisionTensor = Tensor +VisionFrame : TypeAlias = NDArray[Any] +VisionTensor : TypeAlias = Tensor -Batch = Tuple[VisionTensor, VisionTensor] +Batch : TypeAlias = Tuple[VisionTensor, VisionTensor]