use array instead of image
This commit is contained in:
1 parent
a9bc19a27f
commit
453d484526
1 file changed
+6
-4
@@ -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"]]
|
||||
|
||||
Reference in new issue
Block a user