refactor to support lbox-alignerr as a rye workspace

Author: mrobers1982

libs/labelbox/pyproject.toml

@@ -12,7 +12,7 @@ dependencies = [
     "tqdm>=4.66.2",
     "geojson>=3.1.0",
     "lbox-clients==1.1.2",
-    "PyYAML>=6.0",
+    "PyYAML>=6.0"
 ]
 readme = "README.md"
 requires-python = ">=3.9,<3.14"
@@ -56,6 +56,9 @@ data = [
     "typing-extensions>=4.10.0",
     "opencv-python-headless>=4.9.0.80",
 ]
+alignerr = [
+    "lbox-alignerr>=0.1.0",
+]
 
 [build-system]
 requires = ["hatchling"]

libs/labelbox/src/labelbox/__init__.py

@@ -78,7 +78,6 @@
 from labelbox.schema.ontology_kind import OntologyKind
 from labelbox.schema.organization import Organization
 from labelbox.schema.project import Project
-from labelbox.alignerr.schema.project_rate import ProjectRateV2 as ProjectRate
 from labelbox.schema.project_model_config import ProjectModelConfig
 from labelbox.schema.project_overview import (
     ProjectOverview,

libs/labelbox/src/labelbox/alignerr/__init__.py

@@ -82,7 +82,6 @@
 from labelbox.schema.taskstatus import TaskStatus
 from labelbox.schema.api_key import ApiKey
 from labelbox.schema.timeunit import TimeUnit
-from labelbox.alignerr import AlignerrWorkspace
 
 logger = logging.getLogger(__name__)
 
@@ -162,10 +161,6 @@ def enable_experimental(self, value: bool):
     def app_url(self) -> str:
         return self._request_client.app_url
 
-    @property
-    def alignerr_workspace(self) -> AlignerrWorkspace:
-        return AlignerrWorkspace(self)
-
     def set_sdk_method(self, sdk_method: str):
         self._request_client.sdk_method = sdk_method
 

libs/labelbox/src/labelbox/client.py

@@ -395,7 +395,6 @@ class Entity(metaclass=EntityMeta):
         ProjectRole: Type[labelbox.ProjectRole]
         ProjectModelConfig: Type[labelbox.ProjectModelConfig]
         Project: Type[labelbox.Project]
-        ProjectRate: Type[labelbox.ProjectRate]
         Batch: Type[labelbox.Batch]
         CatalogSlice: Type[labelbox.CatalogSlice]
         ModelSlice: Type[labelbox.ModelSlice]

libs/labelbox/src/labelbox/orm/model.py

@@ -0,0 +1,9 @@
+# Alignerr
+
+Alignerr workspace management for Labelbox.
+
+This package provides functionality for managing Alignerr projects, including:
+- Project creation and configuration
+- Rate management for labelers and reviewers
+- Domain and tag management
+- Workforce management

libs/lbox-alignerr/README.md

@@ -0,0 +1,73 @@
+[project]
+name = "lbox-alignerr"
+version = "0.1.0"
+description = "Alignerr workspace management for Labelbox"
+authors = [
+    { name = "Labelbox", email = "engineering@labelbox.com" }
+]
+dependencies = [
+    "labelbox>=0.1.0",
+    "pydantic>=2.0.0",
+    "pyyaml>=6.0",
+]
+readme = "README.md"
+requires-python = ">= 3.9"
+
+classifiers=[
+    # How mature is this project?
+    "Development Status :: 2 - Pre-Alpha",
+    # Indicate who your project is intended for
+    "Topic :: Scientific/Engineering :: Artificial Intelligence",
+    "Topic :: Software Development :: Libraries",
+    "Intended Audience :: Developers",
+    "Intended Audience :: Science/Research",
+    # Pick your license as you wish
+    "License :: OSI Approved :: Apache Software License",
+     # Specify the Python versions you support here.
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+]
+keywords = ["ml", "ai", "labelbox", "labeling", "llm", "machinelearning", "alignerr", "lbox-alignerr"]
+
+[project.urls]
+Homepage = "https://labelbox.com/"
+Documentation = "https://labelbox-python.readthedocs.io/en/latest/"
+Repository = "https://github.com/Labelbox/labelbox-python"
+Issues = "https://github.com/Labelbox/labelbox-python/issues"
+Changelog = "https://github.com/Labelbox/labelbox-python/blob/develop/libs/labelbox/CHANGELOG.md"
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.rye]
+managed = true
+dev-dependencies = [
+    "pytest>=8.1.1",
+    "pytest-cases>=3.8.4",
+    "pytest-rerunfailures>=14.0",
+    "pytest-snapshot>=0.9.0",
+    "pytest-cov>=4.1.0",
+    "pytest-xdist>=3.5.0",
+    "faker>=25.5.0",
+    "pytest-timestamper>=0.0.10",
+    "pytest-timeout>=2.3.1",
+    "pytest-order>=1.2.1",
+    "pyjwt>=2.9.0",
+]
+
+[tool.rye.scripts]
+unit = "pytest tests/unit"
+integration = "pytest tests/integration"
+
+[tool.hatch.metadata]
+allow-direct-references = true
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/alignerr"]
+
+[tool.pytest.ini_options]
+addopts = "-rP -vvv --durations=20 --cov=alignerr --import-mode=importlib"

libs/lbox-alignerr/pyproject.toml

@@ -0,0 +1,3 @@
+from alignerr.alignerr_project import AlignerrWorkspace
+
+__all__ = ["AlignerrWorkspace"]

libs/lbox-alignerr/src/alignerr/__init__.py

@@ -3,13 +3,13 @@
 
 import logging
 
-from labelbox.alignerr.schema.project_rate import ProjectRateV2
-from labelbox.alignerr.schema.project_domain import ProjectDomain
-from labelbox.alignerr.schema.enchanced_resource_tags import (
+from alignerr.schema.project_rate import ProjectRateV2
+from alignerr.schema.project_domain import ProjectDomain
+from alignerr.schema.enchanced_resource_tags import (
     EnhancedResourceTag,
     ResourceTagType,
 )
-from labelbox.alignerr.schema.project_boost_workforce import (
+from alignerr.schema.project_boost_workforce import (
     ProjectBoostWorkforce,
 )
 from labelbox.pagination import PaginatedCollection
@@ -20,7 +20,7 @@
 if TYPE_CHECKING:
     from labelbox import Client
     from labelbox.schema.project import Project
-    from labelbox.alignerr.schema.project_domain import ProjectDomain
+    from alignerr.schema.project_domain import ProjectDomain
 
 
 class AlignerrRole(Enum):
@@ -161,15 +161,19 @@ def __init__(self, client: "Client"):
         self.client = client
 
     def project_builder(self):
-        from labelbox.alignerr.alignerr_project_builder import (
+        from alignerr.alignerr_project_builder import (
             AlignerrProjectBuilder,
         )
 
         return AlignerrProjectBuilder(self.client)
 
     def project_prototype(self):
-        from labelbox.alignerr.alignerr_project_factory import (
+        from alignerr.alignerr_project_factory import (
             AlignerrProjectFactory,
         )
 
         return AlignerrProjectFactory(self.client)
+
+    @classmethod
+    def from_labelbox(cls, client: "Client") -> "AlignerrWorkspace":
+        return cls(client)

libs/labelbox/src/labelbox/alignerr/alignerr_project.py renamed to libs/lbox-alignerr/src/alignerr/alignerr_project.py

@@ -3,14 +3,14 @@
 from typing import TYPE_CHECKING, Optional, Union, List
 import logging
 
-from labelbox.alignerr.schema.project_rate import BillingMode
-from labelbox.alignerr.schema.project_rate import ProjectRateInput
-from labelbox.alignerr.schema.project_domain import ProjectDomain
-from labelbox.alignerr.schema.enchanced_resource_tags import (
+from alignerr.schema.project_rate import BillingMode
+from alignerr.schema.project_rate import ProjectRateInput
+from alignerr.schema.project_domain import ProjectDomain
+from alignerr.schema.enchanced_resource_tags import (
     EnhancedResourceTag,
     ResourceTagType,
 )
-from labelbox.alignerr.schema.project_boost_workforce import (
+from alignerr.schema.project_boost_workforce import (
     ProjectBoostWorkforce,
 )
 from labelbox.schema.media_type import MediaType
@@ -28,7 +28,7 @@ class ValidationType(Enum):
 
 if TYPE_CHECKING:
     from labelbox import Client
-    from labelbox.alignerr.alignerr_project import AlignerrProject, AlignerrRole
+    from alignerr.alignerr_project import AlignerrProject, AlignerrRole
 
 
 class AlignerrProjectBuilder:
@@ -185,7 +185,7 @@ def create(
         labelbox_project = self.client.create_project(**project_data)
 
         # Import here to avoid circular imports
-        from labelbox.alignerr.alignerr_project import AlignerrProject
+        from alignerr.alignerr_project import AlignerrProject
 
         alignerr_project = AlignerrProject(
             self.client, labelbox_project, _internal=True
@@ -254,7 +254,7 @@ def _create_project_owner(self, alignerr_project: "AlignerrProject"):
 
     def _validate_alignerr_rates(self):
         # Import here to avoid circular imports
-        from labelbox.alignerr.alignerr_project import AlignerrRole
+        from alignerr.alignerr_project import AlignerrRole
 
         required_role_rates = set(
             [AlignerrRole.Labeler.value, AlignerrRole.Reviewer.value]