Fixed type

Signed-off-by: Martin <[email protected]>
bmmtstb · Jun 21, 2024 · 09151f2 · 09151f2
1 parent bc0a1e5
commit 09151f2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/dgs/models/dataset/keypoint_rcnn.py b/dgs/models/dataset/keypoint_rcnn.py
@@ -7,6 +7,7 @@
 
 import os
 from abc import ABC
+from typing import Union
 
 import torch as t
 from imagesize import imagesize
@@ -243,7 +244,7 @@ class KeypointRCNNImageBackbone(KeypointRCNNBackbone, ImageDataset):
     __doc__ += KeypointRCNNBackbone.__doc__
 
     data: list[FilePath]
-    masks: list[tvte.Mask | None]
+    masks: list[Union[tvte.Mask, None]]
 
     def __init__(self, config: Config, path: NodePath) -> None:
         KeypointRCNNBackbone.__init__(self, config=config, path=path)