diff --git a/embedding_converter/src/types.py b/embedding_converter/src/types.py index a893b5f..98da9f9 100644 --- a/embedding_converter/src/types.py +++ b/embedding_converter/src/types.py @@ -5,5 +5,5 @@ from torch import Tensor Batch : TypeAlias = Tensor Embedding : TypeAlias = Tensor -Config : TypeAlias = Dict[str, Any] +Config : TypeAlias = Any OptimizerSet : TypeAlias = Any