From a9bc19a27ffe67586e0aa49637a3228f0026b1e9 Mon Sep 17 00:00:00 2001 From: stuartgo Date: Tue, 9 Dec 2025 18:07:20 +0100 Subject: [PATCH] changed to tensor again --- anansyTypes/structures/embedding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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"]]