label-studio-sdk 2.0.7__tar.gz → 2.0.8__tar.gz

{label_studio_sdk-2.0.7 → label_studio_sdk-2.0.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: label-studio-sdk
-Version: 2.0.7
+Version: 2.0.8
 Summary: 
 Requires-Python: >=3.9,<4
 Classifier: Intended Audience :: Developers

{label_studio_sdk-2.0.7 → label_studio_sdk-2.0.8}/pyproject.toml

@@ -3,7 +3,7 @@ name = "label-studio-sdk"
 
 [tool.poetry]
 name = "label-studio-sdk"
-version = "2.0.7"
+version = "2.0.8"
 description = ""
 readme = "README.md"
 authors = []

{label_studio_sdk-2.0.7 → label_studio_sdk-2.0.8}/src/label_studio_sdk/__init__.py

@@ -42,13 +42,14 @@ from .types import (
     Comment,
     CommentRequest,
     CommentSerializerWithExpandedUser,
+    ConfigurablePermissionOption,
+    ConfigurablePermissionOptionDefault,
     ConvertedFormat,
     ConvertedFormatRequest,
     CountLimit,
     CustomScriptsEditableByEnum,
     DefaultRole,
     DefaultRoleCustomScriptsEditableBy,
-    DefaultRoleEnum,
     EditionEnum,
     Export,
     FileUpload,
@@ -81,6 +82,7 @@ from .types import (
     LseOrganizationCustomScriptsEditableBy,
     LseOrganizationMemberList,
     LseProject,
+    LseProjectCounts,
     LseProjectCreate,
     LseProjectCreateSampling,
     LseProjectCreateSkipQueue,
@@ -137,9 +139,12 @@ from .types import (
     OrganizationInvite,
     OrganizationMember,
     OrganizationMembership,
+    OrganizationPermission,
+    OrganizationPermissionRequest,
     PaginatedAllRolesProjectListList,
     PaginatedAnnotationHistoryList,
     PaginatedLseOrganizationMemberListList,
+    PaginatedLseProjectCountsList,
     PaginatedLseUserList,
     PaginatedPaginatedProjectMemberList,
     PaginatedProjectMember,
@@ -290,6 +295,7 @@ from .projects import (
     PatchedLseProjectUpdateRequestSampling,
     PatchedLseProjectUpdateRequestSkipQueue,
     ProjectsDuplicateResponse,
+    ProjectsImportPredictionsResponse,
     ProjectsImportTasksResponse,
     ProjectsListRequestFilter,
 )
@@ -396,13 +402,14 @@ __all__ = [
     "Comment",
     "CommentRequest",
     "CommentSerializerWithExpandedUser",
+    "ConfigurablePermissionOption",
+    "ConfigurablePermissionOptionDefault",
     "ConvertedFormat",
     "ConvertedFormatRequest",
     "CountLimit",
     "CustomScriptsEditableByEnum",
     "DefaultRole",
     "DefaultRoleCustomScriptsEditableBy",
-    "DefaultRoleEnum",
     "EditionEnum",
     "Export",
     "ExportStorageListTypesResponseItem",
@@ -441,6 +448,7 @@ __all__ = [
     "LseOrganizationCustomScriptsEditableBy",
     "LseOrganizationMemberList",
     "LseProject",
+    "LseProjectCounts",
     "LseProjectCreate",
     "LseProjectCreateRequestSampling",
     "LseProjectCreateRequestSkipQueue",
@@ -505,9 +513,12 @@ __all__ = [
     "OrganizationInvite",
     "OrganizationMember",
     "OrganizationMembership",
+    "OrganizationPermission",
+    "OrganizationPermissionRequest",
     "PaginatedAllRolesProjectListList",
     "PaginatedAnnotationHistoryList",
     "PaginatedLseOrganizationMemberListList",
+    "PaginatedLseProjectCountsList",
     "PaginatedLseUserList",
     "PaginatedPaginatedProjectMemberList",
     "PaginatedProjectMember",
@@ -533,6 +544,7 @@ __all__ = [
     "ProjectTemplate",
     "ProjectTemplateRequest",
     "ProjectsDuplicateResponse",
+    "ProjectsImportPredictionsResponse",
     "ProjectsImportTasksResponse",
     "ProjectsListRequestFilter",
     "PromptsCompatibleProjectsRequestProjectType",

label_studio_sdk-2.0.8/src/label_studio_sdk/converter/exports/yolo.py

@@ -0,0 +1,164 @@
+import logging
+import os
+from label_studio_sdk.converter.utils import convert_annotation_to_yolo, convert_annotation_to_yolo_obb
+from label_studio_sdk.converter.keypoints import build_kp_order
+
+logger = logging.getLogger(__name__)
+
+def process_keypoints_for_yolo(labels, label_path,
+                               category_name_to_id, categories,
+                               is_obb, kp_order):
+    class_map = {c['name']: c['id'] for c in categories}
+
+    rectangles = {}
+    for item in labels:
+        if item['type'].lower() == 'rectanglelabels':
+            bbox_id = item['id']
+            cls_name = item['rectanglelabels'][0]
+            cls_idx = class_map.get(cls_name)
+            if cls_idx is None:
+                continue
+
+            x      = item['x'] / 100.0
+            y      = item['y'] / 100.0
+            width  = item['width']  / 100.0
+            height = item['height'] / 100.0
+            x_c    = x + width  / 2.0
+            y_c    = y + height / 2.0
+
+            rectangles[bbox_id] = {
+                'class_idx': cls_idx,
+                'x_center':  x_c,
+                'y_center':  y_c,
+                'width':     width,
+                'height':    height,
+                'kp_dict':   {}
+            }
+
+    for item in labels:
+        if item['type'].lower() == 'keypointlabels':
+            parent_id = item.get('parentID')
+            if parent_id not in rectangles:
+                continue
+            label_name = item['keypointlabels'][0]
+            kp_x = item['x'] / 100.0
+            kp_y = item['y'] / 100.0
+            rectangles[parent_id]['kp_dict'][label_name] = (kp_x, kp_y, 2)  # 2 = visible
+
+    lines = []
+    for rect in rectangles.values():
+        base = [
+            rect['class_idx'],
+            rect['x_center'],
+            rect['y_center'],
+            rect['width'],
+            rect['height']
+        ]
+        keypoints = []
+        for k in kp_order:
+            keypoints.extend(rect['kp_dict'].get(k, (0.0, 0.0, 0)))
+        line = ' '.join(map(str, base + keypoints))
+        lines.append(line)
+
+    with open(label_path, 'w', encoding='utf-8') as f:
+        f.write('\n'.join(lines))
+
+
+def process_and_save_yolo_annotations(labels, label_path, category_name_to_id, categories, is_obb, is_keypoints, label_config):
+    if is_keypoints:
+        kp_order = build_kp_order(label_config)
+        process_keypoints_for_yolo(labels, label_path, category_name_to_id, categories, is_obb, kp_order)
+        return categories, category_name_to_id
+
+    # Stream annotations directly to a temporary file to avoid
+    # accumulating them in memory and to preserve atomic writes.
+    tmp_path = f"{label_path}.tmp"
+
+    try:
+        with open(tmp_path, "w") as f:
+            for label in labels:
+                category_name = None
+                category_names = []  # considering multi-label
+                for key in ["rectanglelabels", "polygonlabels", "labels"]:
+                    if key in label and len(label[key]) > 0:
+                        # change to save multi-label
+                        for category_name in label[key]:
+                            category_names.append(category_name)
+
+                if len(category_names) == 0:
+                    logger.debug(
+                        "Unknown label type or labels are empty: " + str(label)
+                    )
+                    continue
+
+                for category_name in category_names:
+                    if category_name not in category_name_to_id:
+                        category_id = len(categories)
+                        category_name_to_id[category_name] = category_id
+                        categories.append({"id": category_id, "name": category_name})
+                    category_id = category_name_to_id[category_name]
+
+                    if (
+                        "rectanglelabels" in label
+                        or "rectangle" in label
+                        or "labels" in label
+                    ):
+                        # yolo obb
+                        if is_obb:
+                            obb_annotation = convert_annotation_to_yolo_obb(label)
+                            if obb_annotation is None:
+                                continue
+
+                            top_left, top_right, bottom_right, bottom_left = (
+                                obb_annotation
+                            )
+                            x1, y1 = top_left
+                            x2, y2 = top_right
+                            x3, y3 = bottom_right
+                            x4, y4 = bottom_left
+                            annotation_values = [category_id, x1, y1, x2, y2, x3, y3, x4, y4]
+
+                        # simple yolo
+                        else:
+                            annotation = convert_annotation_to_yolo(label)
+                            if annotation is None:
+                                continue
+
+                            (
+                                x,
+                                y,
+                                w,
+                                h,
+                            ) = annotation
+                            annotation_values = [category_id, x, y, w, h]
+
+                    elif "polygonlabels" in label or "polygon" in label:
+                        if not ('points' in label):
+                            continue
+                        points_abs = [(x / 100, y / 100) for x, y in label["points"]]
+                        annotation_values = (
+                            [category_id]
+                            + [coord for point in points_abs for coord in point]
+                        )
+                    else:
+                        raise ValueError(f"Unknown label type {label}")
+
+                    # Write the annotation line immediately
+                    for idx, val in enumerate(annotation_values):
+                        if idx == len(annotation_values) - 1:
+                            f.write(f"{val}\n")
+                        else:
+                            f.write(f"{val} ")
+
+        # Replace the target file atomically after successful write
+        os.replace(tmp_path, label_path)
+
+    finally:
+        # Clean up the temp file in case of exceptions before replace
+        try:
+            if os.path.exists(tmp_path):
+                os.remove(tmp_path)
+        except Exception:
+            pass
+
+    return categories, category_name_to_id

{label_studio_sdk-2.0.7 → label_studio_sdk-2.0.8}/src/label_studio_sdk/core/client_wrapper.py

@@ -4,7 +4,7 @@ import httpx
 
 from .http_client import AsyncHttpClient, HttpClient
 
-VERSION = "2.0.7"
+VERSION = "2.0.8"
 
 class BaseClientWrapper:
     def __init__(

{label_studio_sdk-2.0.7 → label_studio_sdk-2.0.8}/src/label_studio_sdk/import_storage/azure_spi/client.py

@@ -94,6 +94,7 @@ class AzureSpiClient:
         prefix: typing.Optional[str] = OMIT,
         presign: typing.Optional[bool] = OMIT,
         presign_ttl: typing.Optional[int] = OMIT,
+        recursive_scan: typing.Optional[bool] = OMIT,
         regex_filter: typing.Optional[str] = OMIT,
         status: typing.Optional[StatusC5AEnum] = OMIT,
         synchronizable: typing.Optional[bool] = OMIT,
@@ -146,6 +147,9 @@ class AzureSpiClient:
         presign_ttl : typing.Optional[int]
             Presigned URLs TTL (in minutes)
 
+        recursive_scan : typing.Optional[bool]
+            Perform recursive scan
+
         regex_filter : typing.Optional[str]
             Cloud storage regex for filtering objects
 
@@ -204,6 +208,7 @@ class AzureSpiClient:
                 "presign": presign,
                 "presign_ttl": presign_ttl,
                 "project": project,
+                "recursive_scan": recursive_scan,
                 "regex_filter": regex_filter,
                 "status": status,
                 "synchronizable": synchronizable,
@@ -246,6 +251,7 @@ class AzureSpiClient:
         prefix: typing.Optional[str] = OMIT,
         presign: typing.Optional[bool] = OMIT,
         presign_ttl: typing.Optional[int] = OMIT,
+        recursive_scan: typing.Optional[bool] = OMIT,
         regex_filter: typing.Optional[str] = OMIT,
         status: typing.Optional[StatusC5AEnum] = OMIT,
         synchronizable: typing.Optional[bool] = OMIT,
@@ -298,6 +304,9 @@ class AzureSpiClient:
         presign_ttl : typing.Optional[int]
             Presigned URLs TTL (in minutes)
 
+        recursive_scan : typing.Optional[bool]
+            Perform recursive scan
+
         regex_filter : typing.Optional[str]
             Cloud storage regex for filtering objects
 
@@ -355,6 +364,7 @@ class AzureSpiClient:
                 "presign": presign,
                 "presign_ttl": presign_ttl,
                 "project": project,
+                "recursive_scan": recursive_scan,
                 "regex_filter": regex_filter,
                 "status": status,
                 "synchronizable": synchronizable,
@@ -479,6 +489,7 @@ class AzureSpiClient:
         presign: typing.Optional[bool] = OMIT,
         presign_ttl: typing.Optional[int] = OMIT,
         project: typing.Optional[int] = OMIT,
+        recursive_scan: typing.Optional[bool] = OMIT,
         regex_filter: typing.Optional[str] = OMIT,
         status: typing.Optional[StatusC5AEnum] = OMIT,
         synchronizable: typing.Optional[bool] = OMIT,
@@ -533,6 +544,9 @@ class AzureSpiClient:
         project : typing.Optional[int]
             A unique integer value identifying this project.
 
+        recursive_scan : typing.Optional[bool]
+            Perform recursive scan
+
         regex_filter : typing.Optional[str]
             Cloud storage regex for filtering objects
 
@@ -591,6 +605,7 @@ class AzureSpiClient:
                 "presign": presign,
                 "presign_ttl": presign_ttl,
                 "project": project,
+                "recursive_scan": recursive_scan,
                 "regex_filter": regex_filter,
                 "status": status,
                 "synchronizable": synchronizable,
@@ -755,6 +770,7 @@ class AsyncAzureSpiClient:
         prefix: typing.Optional[str] = OMIT,
         presign: typing.Optional[bool] = OMIT,
         presign_ttl: typing.Optional[int] = OMIT,
+        recursive_scan: typing.Optional[bool] = OMIT,
         regex_filter: typing.Optional[str] = OMIT,
         status: typing.Optional[StatusC5AEnum] = OMIT,
         synchronizable: typing.Optional[bool] = OMIT,
@@ -807,6 +823,9 @@ class AsyncAzureSpiClient:
         presign_ttl : typing.Optional[int]
             Presigned URLs TTL (in minutes)
 
+        recursive_scan : typing.Optional[bool]
+            Perform recursive scan
+
         regex_filter : typing.Optional[str]
             Cloud storage regex for filtering objects
 
@@ -873,6 +892,7 @@ class AsyncAzureSpiClient:
                 "presign": presign,
                 "presign_ttl": presign_ttl,
                 "project": project,
+                "recursive_scan": recursive_scan,
                 "regex_filter": regex_filter,
                 "status": status,
                 "synchronizable": synchronizable,
@@ -915,6 +935,7 @@ class AsyncAzureSpiClient:
         prefix: typing.Optional[str] = OMIT,
         presign: typing.Optional[bool] = OMIT,
         presign_ttl: typing.Optional[int] = OMIT,
+        recursive_scan: typing.Optional[bool] = OMIT,
         regex_filter: typing.Optional[str] = OMIT,
         status: typing.Optional[StatusC5AEnum] = OMIT,
         synchronizable: typing.Optional[bool] = OMIT,
@@ -967,6 +988,9 @@ class AsyncAzureSpiClient:
         presign_ttl : typing.Optional[int]
             Presigned URLs TTL (in minutes)
 
+        recursive_scan : typing.Optional[bool]
+            Perform recursive scan
+
         regex_filter : typing.Optional[str]
             Cloud storage regex for filtering objects
 
@@ -1032,6 +1056,7 @@ class AsyncAzureSpiClient:
                 "presign": presign,
                 "presign_ttl": presign_ttl,
                 "project": project,
+                "recursive_scan": recursive_scan,
                 "regex_filter": regex_filter,
                 "status": status,
                 "synchronizable": synchronizable,
@@ -1172,6 +1197,7 @@ class AsyncAzureSpiClient:
         presign: typing.Optional[bool] = OMIT,
         presign_ttl: typing.Optional[int] = OMIT,
         project: typing.Optional[int] = OMIT,
+        recursive_scan: typing.Optional[bool] = OMIT,
         regex_filter: typing.Optional[str] = OMIT,
         status: typing.Optional[StatusC5AEnum] = OMIT,
         synchronizable: typing.Optional[bool] = OMIT,
@@ -1226,6 +1252,9 @@ class AsyncAzureSpiClient:
         project : typing.Optional[int]
             A unique integer value identifying this project.
 
+        recursive_scan : typing.Optional[bool]
+            Perform recursive scan
+
         regex_filter : typing.Optional[str]
             Cloud storage regex for filtering objects
 
@@ -1292,6 +1321,7 @@ class AsyncAzureSpiClient:
                 "presign": presign,
                 "presign_ttl": presign_ttl,
                 "project": project,
+                "recursive_scan": recursive_scan,
                 "regex_filter": regex_filter,
                 "status": status,
                 "synchronizable": synchronizable,

{label_studio_sdk-2.0.7 → label_studio_sdk-2.0.8}/src/label_studio_sdk/organizations/__init__.py

@@ -1,6 +1,6 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from .types import PatchedDefaultRoleRequestCustomScriptsEditableBy
-from . import invites, members
+from . import invites, members, permissions
 
-__all__ = ["PatchedDefaultRoleRequestCustomScriptsEditableBy", "invites", "members"]
+__all__ = ["PatchedDefaultRoleRequestCustomScriptsEditableBy", "invites", "members", "permissions"]

{label_studio_sdk-2.0.7 → label_studio_sdk-2.0.8}/src/label_studio_sdk/organizations/client.py

@@ -4,6 +4,7 @@ import typing
 from ..core.client_wrapper import SyncClientWrapper
 from .invites.client import InvitesClient
 from .members.client import MembersClient
+from .permissions.client import PermissionsClient
 from ..core.request_options import RequestOptions
 from ..types.organization_invite import OrganizationInvite
 from ..core.unchecked_base_model import construct_type
@@ -19,12 +20,13 @@ import datetime as dt
 from .types.patched_default_role_request_custom_scripts_editable_by import (
     PatchedDefaultRoleRequestCustomScriptsEditableBy,
 )
-from ..types.default_role_enum import DefaultRoleEnum
+from ..types.role9e7enum import Role9E7Enum
 from ..types.default_role import DefaultRole
 from ..core.serialization import convert_and_respect_annotation_metadata
 from ..core.client_wrapper import AsyncClientWrapper
 from .invites.client import AsyncInvitesClient
 from .members.client import AsyncMembersClient
+from .permissions.client import AsyncPermissionsClient
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -35,6 +37,7 @@ class OrganizationsClient:
         self._client_wrapper = client_wrapper
         self.invites = InvitesClient(client_wrapper=self._client_wrapper)
         self.members = MembersClient(client_wrapper=self._client_wrapper)
+        self.permissions = PermissionsClient(client_wrapper=self._client_wrapper)
 
     def reset_token(self, *, request_options: typing.Optional[RequestOptions] = None) -> OrganizationInvite:
         """
@@ -306,7 +309,7 @@ class OrganizationsClient:
         annotator_reviewer_firewall_enabled_at: typing.Optional[dt.datetime] = OMIT,
         custom_scripts_editable_by: typing.Optional[PatchedDefaultRoleRequestCustomScriptsEditableBy] = OMIT,
         custom_scripts_enabled_at: typing.Optional[dt.datetime] = OMIT,
-        default_role: typing.Optional[DefaultRoleEnum] = OMIT,
+        default_role: typing.Optional[Role9E7Enum] = OMIT,
         email_notification_settings: typing.Optional[typing.Optional[typing.Any]] = OMIT,
         embed_domains: typing.Optional[typing.Optional[typing.Any]] = OMIT,
         embed_settings: typing.Optional[typing.Optional[typing.Any]] = OMIT,
@@ -336,7 +339,7 @@ class OrganizationsClient:
         custom_scripts_enabled_at : typing.Optional[dt.datetime]
             Set to current time to enabled custom scripts for this organization. Can only be enabled if no organization members are active members of any other organizations; otherwise an error will be raised. If this occurs, contact the LEAP team for assistance with enabling custom scripts.
 
-        default_role : typing.Optional[DefaultRoleEnum]
+        default_role : typing.Optional[Role9E7Enum]
             Default membership role for invited users
 
             * `OW` - Owner
@@ -433,6 +436,7 @@ class AsyncOrganizationsClient:
         self._client_wrapper = client_wrapper
         self.invites = AsyncInvitesClient(client_wrapper=self._client_wrapper)
         self.members = AsyncMembersClient(client_wrapper=self._client_wrapper)
+        self.permissions = AsyncPermissionsClient(client_wrapper=self._client_wrapper)
 
     async def reset_token(self, *, request_options: typing.Optional[RequestOptions] = None) -> OrganizationInvite:
         """
@@ -736,7 +740,7 @@ class AsyncOrganizationsClient:
         annotator_reviewer_firewall_enabled_at: typing.Optional[dt.datetime] = OMIT,
         custom_scripts_editable_by: typing.Optional[PatchedDefaultRoleRequestCustomScriptsEditableBy] = OMIT,
         custom_scripts_enabled_at: typing.Optional[dt.datetime] = OMIT,
-        default_role: typing.Optional[DefaultRoleEnum] = OMIT,
+        default_role: typing.Optional[Role9E7Enum] = OMIT,
         email_notification_settings: typing.Optional[typing.Optional[typing.Any]] = OMIT,
         embed_domains: typing.Optional[typing.Optional[typing.Any]] = OMIT,
         embed_settings: typing.Optional[typing.Optional[typing.Any]] = OMIT,
@@ -766,7 +770,7 @@ class AsyncOrganizationsClient:
         custom_scripts_enabled_at : typing.Optional[dt.datetime]
             Set to current time to enabled custom scripts for this organization. Can only be enabled if no organization members are active members of any other organizations; otherwise an error will be raised. If this occurs, contact the LEAP team for assistance with enabling custom scripts.
 
-        default_role : typing.Optional[DefaultRoleEnum]
+        default_role : typing.Optional[Role9E7Enum]
             Default membership role for invited users
 
             * `OW` - Owner