Add Project get_labeling_service_status() (#1753)

Author: vbrodsky

libs/labelbox/src/labelbox/schema/project.py

@@ -9,7 +9,7 @@
 from typing import TYPE_CHECKING, Any, Dict, Iterable, List, Optional, Tuple, TypeVar, Union, overload
 from urllib.parse import urlparse
 
-from labelbox.schema.labeling_service import LabelingService
+from labelbox.schema.labeling_service import LabelingService, LabelingServiceStatus
 import requests
 
 from labelbox import parser
@@ -1920,10 +1920,25 @@ def clone(self) -> "Project":
     def get_labeling_service(self) -> LabelingService:
         """Get the labeling service for this project.
 
+        Raises:
+            ResourceNotFoundError if the project does not have a labeling service.
+
         Returns:
             LabelingService: The labeling service for this project.
         """
-        return LabelingService.get(self.client, self.uid)  # type: ignore
+        return LabelingService.get(self.client, self.uid)
+
+    @experimental
+    def get_labeling_service_status(self) -> LabelingServiceStatus:
+        """Get the labeling service status for this project.
+
+        Raises:
+            ResourceNotFoundError if the project does not have a labeling service.
+
+        Returns:
+            LabelingServiceStatus: The labeling service status for this project.
+        """
+        return self.get_labeling_service().status
 
     @experimental
     def request_labeling_service(self) -> LabelingService:

libs/labelbox/tests/integration/test_labeling_service.py

@@ -7,6 +7,8 @@
 def test_get_labeling_service_throws_exception(project):
     with pytest.raises(ResourceNotFoundError):  # No labeling service by default
         project.get_labeling_service()
+    with pytest.raises(ResourceNotFoundError):  # No labeling service by default
+        project.get_labeling_service_status()
 
 
 def test_start_labeling_service(project):
@@ -18,3 +20,6 @@ def test_start_labeling_service(project):
     labeling_service = project.get_labeling_service()
     assert labeling_service.status == LabelingServiceStatus.SetUp
     assert labeling_service.project_id == project.uid
+
+    labeling_service_status = project.get_labeling_service_status()
+    assert labeling_service_status == LabelingServiceStatus.SetUp