diff --git a/anansyTypes/structures/embedding.py b/anansyTypes/structures/embedding.py index 3d3f256..7eb1d20 100644 --- a/anansyTypes/structures/embedding.py +++ b/anansyTypes/structures/embedding.py @@ -6,7 +6,7 @@ import torch from typing import Any, Literal @dataclass(frozen=True) class Embedding: - data: Tensor[torch.Tensor | np.ndarray[Any, Any], Literal["int32", "int64"]] + data: Tensor[torch.Tensor | np.ndarray[Any, Any],tuple, Literal["int32", "int64"]]