Skip to content

Commit ae0f625

Browse files
committed
fix type annotation Image -> Optional[Image.Image]
1 parent 4311757 commit ae0f625

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

src/image/image.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import io
2+
from typing import Optional
23

34
import numpy as np
45
import torch
@@ -18,7 +19,12 @@ def get_tensor_from_np_array(np_array: np.ndarray) -> torch.Tensor:
1819

1920

2021
class ImageObject:
21-
def __init__(self, viam_image: ViamImage, pil_image: Image = None, device=None):
22+
def __init__(
23+
self,
24+
viam_image: ViamImage,
25+
pil_image: Optional[Image.Image] = None,
26+
device=None
27+
):
2228
self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
2329
if pil_image is not None:
2430
self.pil_image = pil_image

0 commit comments

Comments
 (0)