From a9eea6487808d0ff98e9fb45b3fcb04b950c5e17 Mon Sep 17 00:00:00 2001
From: stuartgo <ottersenstuart@gmail.com>
Date: Tue, 9 Dec 2025 18:05:37 +0100
Subject: [PATCH] changed to tensor again

---
 anansyTypes/structures/embedding.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/anansyTypes/structures/embedding.py b/anansyTypes/structures/embedding.py
index 176c32d..3d3f256 100644
--- a/anansyTypes/structures/embedding.py
+++ b/anansyTypes/structures/embedding.py
@@ -3,9 +3,10 @@ from dataclasses import dataclass
 import numpy as np
 from pydantic_tensor import Tensor
 import torch
+from typing import Any, Literal
 @dataclass(frozen=True)
 class Embedding:
-    data: Tensor[torch.Tensor]
+    data: Tensor[torch.Tensor | np.ndarray[Any, Any], Literal["int32", "int64"]]