Merge pull request #66 from JigsawStack/feat/object-detection-feature

typed now as object instead of object detection

Author: Khurdhula-Harshavardhan

jigsawstack/vision.py

@@ -102,9 +102,9 @@ class ObjectDetectionParams(TypedDict):
     List of prompts for object detection
     """
 
-    features: NotRequired[List[Literal["object_detection", "gui"]]]
+    features: NotRequired[List[Literal["object", "gui"]]]
     """
-    List of features to enable: object_detection, gui
+    List of features to enable: object, gui
     """
 
     annotated_image: NotRequired[bool]

tests/test_object_detection.py

@@ -49,7 +49,7 @@
         "name": "with_blob_both_features",
         "blob": IMAGE_URL,
         "options": {
-            "features": ["object_detection", "gui"],
+            "features": ["object", "gui"],
             "annotated_image": True,
             "return_type": "url",
         },
@@ -63,7 +63,7 @@
         "name": "with_blob_object_detection_features",
         "blob": IMAGE_URL,
         "options": {
-            "features": ["object_detection"],
+            "features": ["object"],
             "annotated_image": True,
             "return_type": "base64",
         },
@@ -80,7 +80,7 @@
         "name": "with_all_options",
         "blob": IMAGE_URL,
         "options": {
-            "features": ["object_detection", "gui"],
+            "features": ["object", "gui"],
             "prompts": ["car", "road", "tree"],
             "annotated_image": True,
             "return_type": "base64",