label-studio-sdk 1.0.5__py3-none-any.whl → 1.0.8__py3-none-any.whl

label_studio_sdk/comments/client.py

@@ -0,0 +1,512 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from json.decoder import JSONDecodeError
+
+from ..core.api_error import ApiError
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.jsonable_encoder import jsonable_encoder
+from ..core.pydantic_utilities import pydantic_v1
+from ..core.request_options import RequestOptions
+from ..types.comment import Comment
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class CommentsClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def list(
+        self,
+        *,
+        project: typing.Optional[int] = None,
+        expand_created_by: typing.Optional[bool] = None,
+        annotation: typing.Optional[int] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.List[Comment]:
+        """
+        Get a list of comments for a specific project.
+
+        Parameters
+        ----------
+        project : typing.Optional[int]
+            Project ID
+
+        expand_created_by : typing.Optional[bool]
+            Expand the created_by field with object instead of ID
+
+        annotation : typing.Optional[int]
+            Annotation ID
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[Comment]
+
+
+        Examples
+        --------
+        from label_studio_sdk.client import LabelStudio
+
+        client = LabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        client.comments.list()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/comments/",
+            method="GET",
+            params={"project": project, "expand_created_by": expand_created_by, "annotation": annotation},
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(typing.List[Comment], _response.json())  # type: ignore
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def create(
+        self,
+        *,
+        annotation: typing.Optional[int] = OMIT,
+        project: typing.Optional[int] = OMIT,
+        text: typing.Optional[str] = OMIT,
+        is_resolved: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> Comment:
+        """
+        Create a new comment.
+
+        Parameters
+        ----------
+        annotation : typing.Optional[int]
+
+        project : typing.Optional[int]
+
+        text : typing.Optional[str]
+
+        is_resolved : typing.Optional[bool]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        Comment
+
+
+        Examples
+        --------
+        from label_studio_sdk.client import LabelStudio
+
+        client = LabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        client.comments.create()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/comments/",
+            method="POST",
+            json={"annotation": annotation, "project": project, "text": text, "is_resolved": is_resolved},
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(Comment, _response.json())  # type: ignore
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def get(self, id: int, *, request_options: typing.Optional[RequestOptions] = None) -> Comment:
+        """
+        Get a specific comment.
+
+        Parameters
+        ----------
+        id : int
+            Comment ID
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        Comment
+
+
+        Examples
+        --------
+        from label_studio_sdk.client import LabelStudio
+
+        client = LabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        client.comments.get(
+            id=1,
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"api/comments/{jsonable_encoder(id)}", method="GET", request_options=request_options
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(Comment, _response.json())  # type: ignore
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def delete(self, id: int, *, request_options: typing.Optional[RequestOptions] = None) -> None:
+        """
+        Delete a specific comment.
+
+        Parameters
+        ----------
+        id : int
+            Comment ID
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from label_studio_sdk.client import LabelStudio
+
+        client = LabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        client.comments.delete(
+            id=1,
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"api/comments/{jsonable_encoder(id)}", method="DELETE", request_options=request_options
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def update(
+        self,
+        id: int,
+        *,
+        annotation: typing.Optional[int] = OMIT,
+        project: typing.Optional[int] = OMIT,
+        text: typing.Optional[str] = OMIT,
+        is_resolved: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> Comment:
+        """
+        Update a specific comment.
+
+        Parameters
+        ----------
+        id : int
+            Comment ID
+
+        annotation : typing.Optional[int]
+
+        project : typing.Optional[int]
+
+        text : typing.Optional[str]
+
+        is_resolved : typing.Optional[bool]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        Comment
+
+
+        Examples
+        --------
+        from label_studio_sdk.client import LabelStudio
+
+        client = LabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        client.comments.update(
+            id=1,
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"api/comments/{jsonable_encoder(id)}",
+            method="PATCH",
+            json={"annotation": annotation, "project": project, "text": text, "is_resolved": is_resolved},
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(Comment, _response.json())  # type: ignore
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+
+class AsyncCommentsClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def list(
+        self,
+        *,
+        project: typing.Optional[int] = None,
+        expand_created_by: typing.Optional[bool] = None,
+        annotation: typing.Optional[int] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.List[Comment]:
+        """
+        Get a list of comments for a specific project.
+
+        Parameters
+        ----------
+        project : typing.Optional[int]
+            Project ID
+
+        expand_created_by : typing.Optional[bool]
+            Expand the created_by field with object instead of ID
+
+        annotation : typing.Optional[int]
+            Annotation ID
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[Comment]
+
+
+        Examples
+        --------
+        from label_studio_sdk.client import AsyncLabelStudio
+
+        client = AsyncLabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        await client.comments.list()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/comments/",
+            method="GET",
+            params={"project": project, "expand_created_by": expand_created_by, "annotation": annotation},
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(typing.List[Comment], _response.json())  # type: ignore
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def create(
+        self,
+        *,
+        annotation: typing.Optional[int] = OMIT,
+        project: typing.Optional[int] = OMIT,
+        text: typing.Optional[str] = OMIT,
+        is_resolved: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> Comment:
+        """
+        Create a new comment.
+
+        Parameters
+        ----------
+        annotation : typing.Optional[int]
+
+        project : typing.Optional[int]
+
+        text : typing.Optional[str]
+
+        is_resolved : typing.Optional[bool]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        Comment
+
+
+        Examples
+        --------
+        from label_studio_sdk.client import AsyncLabelStudio
+
+        client = AsyncLabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        await client.comments.create()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/comments/",
+            method="POST",
+            json={"annotation": annotation, "project": project, "text": text, "is_resolved": is_resolved},
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(Comment, _response.json())  # type: ignore
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def get(self, id: int, *, request_options: typing.Optional[RequestOptions] = None) -> Comment:
+        """
+        Get a specific comment.
+
+        Parameters
+        ----------
+        id : int
+            Comment ID
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        Comment
+
+
+        Examples
+        --------
+        from label_studio_sdk.client import AsyncLabelStudio
+
+        client = AsyncLabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        await client.comments.get(
+            id=1,
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"api/comments/{jsonable_encoder(id)}", method="GET", request_options=request_options
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(Comment, _response.json())  # type: ignore
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def delete(self, id: int, *, request_options: typing.Optional[RequestOptions] = None) -> None:
+        """
+        Delete a specific comment.
+
+        Parameters
+        ----------
+        id : int
+            Comment ID
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from label_studio_sdk.client import AsyncLabelStudio
+
+        client = AsyncLabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        await client.comments.delete(
+            id=1,
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"api/comments/{jsonable_encoder(id)}", method="DELETE", request_options=request_options
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def update(
+        self,
+        id: int,
+        *,
+        annotation: typing.Optional[int] = OMIT,
+        project: typing.Optional[int] = OMIT,
+        text: typing.Optional[str] = OMIT,
+        is_resolved: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> Comment:
+        """
+        Update a specific comment.
+
+        Parameters
+        ----------
+        id : int
+            Comment ID
+
+        annotation : typing.Optional[int]
+
+        project : typing.Optional[int]
+
+        text : typing.Optional[str]
+
+        is_resolved : typing.Optional[bool]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        Comment
+
+
+        Examples
+        --------
+        from label_studio_sdk.client import AsyncLabelStudio
+
+        client = AsyncLabelStudio(
+            api_key="YOUR_API_KEY",
+        )
+        await client.comments.update(
+            id=1,
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"api/comments/{jsonable_encoder(id)}",
+            method="PATCH",
+            json={"annotation": annotation, "project": project, "text": text, "is_resolved": is_resolved},
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return pydantic_v1.parse_obj_as(Comment, _response.json())  # type: ignore
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)

label_studio_sdk/converter/converter.py

@@ -10,7 +10,6 @@ from copy import deepcopy
 from datetime import datetime
 from enum import Enum
 from glob import glob
-from operator import itemgetter
 from shutil import copy2
 from typing import Optional
 
@@ -282,12 +281,16 @@ class Converter(object):
         for name, info in self._schema.items():
             if output_tags is not None and name not in output_tags:
                 continue
-            data_keys |= set(map(itemgetter("value"), info["inputs"]))
+            for input_tag in info["inputs"]:
+                for value_key_name in ["value", "valueList"]:
+                    if value_key_name in input_tag:
+                        data_keys.add(input_tag[value_key_name])
             output_tag_names.append(name)
 
         return list(data_keys), output_tag_names
 
     def _get_supported_formats(self):
+        is_mig = False
         if len(self._data_keys) > 1:
             return [
                 Format.JSON.name,
@@ -300,6 +303,8 @@ class Converter(object):
         for info in self._schema.values():
             output_tag_types.add(info["type"])
             for input_tag in info["inputs"]:
+                if input_tag.get("valueList"):
+                    is_mig = True
                 if input_tag["type"] == "Text" and input_tag.get("valueType") == "url":
                     logger.error('valueType="url" are not supported for text inputs')
                     continue
@@ -308,7 +313,7 @@ class Converter(object):
         all_formats = [f.name for f in Format]
         if not ("Text" in input_tag_types and "Labels" in output_tag_types):
             all_formats.remove(Format.CONLL2003.name)
-        if not (
+        if is_mig or not (
             "Image" in input_tag_types
             and (
                 "RectangleLabels" in output_tag_types
@@ -317,7 +322,7 @@ class Converter(object):
             )
         ):
             all_formats.remove(Format.VOC.name)
-        if not (
+        if is_mig or not (
             "Image" in input_tag_types
             and (
                 "RectangleLabels" in output_tag_types
@@ -346,6 +351,8 @@ class Converter(object):
             and "TextArea" in output_tag_types
         ):
             all_formats.remove(Format.ASR_MANIFEST.name)
+        if is_mig or ('Video' in input_tag_types and 'TimelineLabels' in output_tag_types):
+            all_formats.remove(Format.YOLO_OBB.name)
 
         return all_formats
 

label_studio_sdk/converter/imports/coco.py

@@ -48,10 +48,9 @@ def create_bbox(annotation, categories, from_name, image_height, image_width, to
 
 
 def create_segmentation(
-    annotation, categories, from_name, image_height, image_width, to_name
+    category_id, segmentation, categories, from_name, image_height, image_width, to_name
 ):
-    label = categories[int(annotation["category_id"])]
-    segmentation = annotation["segmentation"][0]
+    label = categories[int(category_id)]
     points = [list(x) for x in zip(*[iter(segmentation)] * 2)]
 
     for i in range(len(points)):
@@ -216,15 +215,17 @@ def convert_coco_to_ls(
             task[out_type][0]["result"].append(item)
 
         if "segmentation" in annotation and len(annotation["segmentation"]):
-            item = create_segmentation(
-                annotation,
-                categories,
-                segmentation_from_name,
-                image_height,
-                image_width,
-                to_name,
-            )
-            task[out_type][0]["result"].append(item)
+            for single_segmentation in annotation["segmentation"]:
+                item = create_segmentation(
+                    annotation["category_id"],
+                    single_segmentation,
+                    categories,
+                    segmentation_from_name,
+                    image_height,
+                    image_width,
+                    to_name,
+                )
+                task[out_type][0]["result"].append(item)
 
         if "keypoints" in annotation:
             items = create_keypoints(
@@ -270,7 +271,7 @@ def add_parser(subparsers):
         "--input",
         dest="input",
         required=True,
-        help="directory with COCO where images, labels, notes.json are located",
+        help="input COCO json file",
         action=ExpandFullPath,
     )
     coco.add_argument(