Merge pull request #23 from epoch8/0.3.4

0.3.4

Author: bobokvsky

.github/workflows/pytest.yaml

@@ -11,7 +11,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.10", "3.11"]
+        python-version: ["3.9", "3.10", "3.11"]
 
     steps:
       - uses: actions/checkout@v2

CHANGELOG.md

@@ -1,3 +1,6 @@
+# 0.3.4
+* Update label_studio_sdk dependency (allow >2 pydantic)
+
 # 0.3.3
 * Update python dependency (allow 3.11 python)
 

datapipe_label_studio_lite/pipeline.py

@@ -26,7 +26,7 @@
 from datapipe.step.datatable_transform import DatatableTransformStep
 from datapipe.datatable import DataTable
 from datapipe.store.database import DBConn
-import label_studio_sdk
+import label_studio_sdk._legacy as label_studio_sdk
 from datapipe_label_studio_lite.sdk_utils import get_project_by_title, get_tasks_iter
 from label_studio_sdk.data_manager import Filters, Operator, Type, DATETIME_FORMAT
 
@@ -103,12 +103,12 @@ def project(self) -> label_studio_sdk.Project:
         if self._project is not None:
             return self._project
         assert self.ls_client.check_connection(), "No connection to LS."
-        self._project = (
+        _project = (
             self.project_identifier
             if str(self.project_identifier).isnumeric()
             else get_project_by_title(self.ls_client, str(self.project_identifier))
         )
-        if self._project is None:
+        if _project is None:
             self._project = self.ls_client.start_project(
                 title=self.project_identifier,
                 description=self.project_description_at_create,
@@ -134,6 +134,9 @@ def project(self) -> label_studio_sdk.Project:
                 task_data_password=None,
                 control_weights={},
             )
+        else:
+            assert isinstance(_project, label_studio_sdk.Project)
+            self._project = _project
 
         return self._project
 

datapipe_label_studio_lite/sdk_utils.py

@@ -1,5 +1,5 @@
 import requests
-import label_studio_sdk
+import label_studio_sdk._legacy as label_studio_sdk
 from urllib.parse import urljoin
 from typing import Any, Dict, Iterator, List, Optional
 

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "datapipe-label-studio-lite"
-version = "0.3.3"
+version = "0.3.4"
 description = ""
 authors = [
     "Alexander Kozlov <bobokvsky@epoch8.co>",
@@ -11,12 +11,9 @@ exclude = ["tests", "experiments"]
 
 [tool.poetry.dependencies]
 python = ">=3.8, <3.12"
-pandas = ">=1.2.0, <3"
-numpy = "^1.21.0"
-SQLAlchemy = "^1.4.25"
-requests = "^2.24.0"
+requests = ">=2.24.0, <3"
 datapipe-core = ">=0.13.0, <0.14"
-label-studio-sdk = "^0.0.16"
+label-studio-sdk = ">=1.0.3, <1.1"
 
 [tool.poetry.dev-dependencies]
 flake8 = "*"