fix: typehints.

jws-1 · Mar 8, 2024 · febe5fa · febe5fa
1 parent 6b9f223
commit febe5fa
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/skills/src/lasr_skills/detect.py b/skills/src/lasr_skills/detect.py
@@ -6,7 +6,7 @@
 
 from lasr_vision_msgs.srv import YoloDetection
 
-from typing import List
+from typing import List, Union
 
 
 class Detect(smach.State):
@@ -15,7 +15,7 @@ def __init__(
         self,
         image_topic: str = "/xtion/rgb/image_raw",
         model: str = "yolov8n.pt",
-        filter: List[str] | None = None,
+        filter: Union[List[str], None] = None,
         confidence: float = 0.5,
         nms: float = 0.3,
     ):

diff --git a/skills/src/lasr_skills/detect_3d.py b/skills/src/lasr_skills/detect_3d.py
@@ -4,7 +4,7 @@
 from sensor_msgs.msg import PointCloud2
 from lasr_vision_msgs.srv import YoloDetection3D
 
-from typing import List
+from typing import List, Union
 
 
 class Detect3D(smach.State):
@@ -13,7 +13,7 @@ def __init__(
         self,
         depth_topic: str = "/xtion/depth_registered/points",
         model: str = "yolov8n-seg.pt",
-        filter: List[str] | None = None,
+        filter: Union[List[str], None] = None,
         confidence: float = 0.5,
         nms: float = 0.3,
     ):

diff --git a/skills/src/lasr_skills/detect_3d_in_area.py b/skills/src/lasr_skills/detect_3d_in_area.py
@@ -2,7 +2,7 @@
 
 from lasr_skills import Detect3D
 
-from typing import List
+from typing import List, Union
 
 from shapely.geometry import Point
 from shapely.geometry.polygon import Polygon
@@ -40,7 +40,7 @@ def __init__(
         area_polygon: Polygon,
         depth_topic: str = "/xtion/depth_registered/points",
         model: str = "yolov8n-seg.pt",
-        filter: List[str] | None = None,
+        filter: Union[List[str], None] = None,
         confidence: float = 0.5,
         nms: float = 0.3,
     ):