From c9b0c86db89e6a56f7ca697691f36ecb3fa155aa Mon Sep 17 00:00:00 2001 From: Val Brodsky Date: Mon, 15 Jul 2024 15:40:30 -0400 Subject: [PATCH 1/2] Remove project Json media type --- libs/labelbox/src/labelbox/schema/media_type.py | 1 - 1 file changed, 1 deletion(-) diff --git a/libs/labelbox/src/labelbox/schema/media_type.py b/libs/labelbox/src/labelbox/schema/media_type.py index 88522c335..22a66ff5e 100644 --- a/libs/labelbox/src/labelbox/schema/media_type.py +++ b/libs/labelbox/src/labelbox/schema/media_type.py @@ -9,7 +9,6 @@ class MediaType(Enum): Geospatial_Tile = "TMS_GEO" Html = "HTML" Image = "IMAGE" - Json = "JSON" LLMPromptCreation = "LLM_PROMPT_CREATION" LLMPromptResponseCreation = "LLM_PROMPT_RESPONSE_CREATION" Pdf = "PDF" From a1b1d485f36eaa0ca4c751b85ffeca0b851e7ca4 Mon Sep 17 00:00:00 2001 From: Val Brodsky Date: Mon, 15 Jul 2024 16:05:03 -0400 Subject: [PATCH 2/2] Fix labeling front end integration test --- libs/labelbox/src/labelbox/client.py | 1 - .../tests/integration/test_labeling_frontend.py | 10 ++++++++-- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/libs/labelbox/src/labelbox/client.py b/libs/labelbox/src/labelbox/client.py index 6dbe9d60e..1e59ca023 100644 --- a/libs/labelbox/src/labelbox/client.py +++ b/libs/labelbox/src/labelbox/client.py @@ -919,7 +919,6 @@ def _create_project(self, **kwargs) -> Project: params.pop("append_to_existing_dataset", None), } extra_params = {k: v for k, v in extra_params.items() if v is not None} - return self._create(Entity.Project, params, extra_params) def get_roles(self) -> List[Role]: diff --git a/libs/labelbox/tests/integration/test_labeling_frontend.py b/libs/labelbox/tests/integration/test_labeling_frontend.py index da6170708..d91bac8ba 100644 --- a/libs/labelbox/tests/integration/test_labeling_frontend.py +++ b/libs/labelbox/tests/integration/test_labeling_frontend.py @@ -8,12 +8,18 @@ def test_get_labeling_frontends(client): def test_labeling_frontend_connecting_to_project(project): - assert project.labeling_frontend() == None + client = project.client + default_labeling_frontend = next( + client.get_labeling_frontends(where=LabelingFrontend.name == "Editor")) + + assert project.labeling_frontend( + ) == default_labeling_frontend # we now have a default labeling frontend frontend = list(project.client.get_labeling_frontends())[0] + project.labeling_frontend.connect(frontend) project.labeling_frontend.connect(frontend) - assert project.labeling_frontend() == frontend + assert project.labeling_frontend() == default_labeling_frontend project.labeling_frontend.disconnect(frontend) assert project.labeling_frontend() == None