use array instead of image

This commit is contained in:
stuart committed 2025-12-09 18:12:00 +01:00
1 parent a9bc19a27f
commit 453d484526
1 file changed
+6 -4
+6 -4
View File
@@ -1,10 +1,12 @@
from dataclasses import dataclass
from typing import Optional
from PIL import Image
import numpy as np
from pydantic_tensor import Tensor
from typing import Any, Literal
import torch
@dataclass
class Symbol:
id: int
label: str
image: Optional[Image.Image]
image: Tensor[ np.ndarray[Any, Any],tuple, Literal["int32", "int64"]]