Cleanup LabelingDashboard

Author: Val Brodsky

libs/labelbox/src/labelbox/__init__.py

@@ -43,4 +43,6 @@
 from labelbox.schema.identifiable import UniqueId, GlobalKey
 from labelbox.schema.ontology_kind import OntologyKind
 from labelbox.schema.project_overview import ProjectOverview, ProjectOverviewDetailed
-from labelbox.schema.labeling_service import LabelingService, LabelingServiceStatus
+from labelbox.schema.labeling_service import LabelingService
+from labelbox.schema.labeling_service_dashboard import LabelingServiceDashboard
+from labelbox.schema.labeling_service_status import LabelingServiceStatus

libs/labelbox/src/labelbox/client.py

@@ -2397,4 +2397,11 @@ def get_labeling_service_dashboards(
         after: Optional[str] = None,
         where: Optional[Comparison] = None,
     ) -> PaginatedCollection:
+        """
+        Get all labeling service dashboards for a given org.
+
+        Optional parameters:
+            after: The cursor to use for pagination.
+            where: A filter to apply to the query.
+        """
         return LabelingServiceDashboard.get_all(self, after, where)

libs/labelbox/src/labelbox/schema/labeling_service_dashboard.py

@@ -1,14 +1,18 @@
-from datetime import datetime
+from email.policy import default
 from string import Template
+from datetime import datetime
 from typing import Any, Dict, List, Optional
 
+import sys
+
+if sys.version_info[:2] == (3, 8):
+    from typing_extensions import Annotated
+else:
+    from typing import Annotated
+
 from labelbox.exceptions import ResourceNotFoundError
-from labelbox.orm.comparison import Comparison
-from labelbox.orm import query
-from ..orm.model import Field
 from labelbox.pagination import PaginatedCollection
-from labelbox.pydantic_compat import BaseModel, root_validator
-from .organization import Organization
+from labelbox.pydantic_compat import BaseModel, root_validator, Field
 from labelbox.utils import _CamelCaseMixin
 from labelbox.schema.labeling_service_status import LabelingServiceStatus
 
@@ -28,15 +32,29 @@
 
 
 class LabelingServiceDashboard(BaseModel):
-    id: str
-    name: str
-    # service_type: str
-    # created_at: datetime
-    # updated_at: datetime
-    # created_by_id: str
-    status: LabelingServiceStatus
-    tasks_completed: int
-    tasks_remaining: int
+    """
+    Represent labeling service data for a project
+
+    Attributes:
+        id (str): project id
+        name (str): project name
+        status (LabelingServiceStatus): status of the labeling service
+        tasks_completed (int): number of data rows completed
+        tasks_remaining (int): number of data rows that have not started
+        client (Any): labelbox client
+    """
+    id: str = Field(frozen=True)
+    name: str = Field(frozen=True)
+    service_type: Optional[str] = Field(frozen=True, default=None)
+    created_at: Optional[datetime] = Field(frozen=True, default=None)
+    updated_at: Optional[datetime] = Field(frozen=True, default=None)
+    created_by_id: Optional[str] = Field(frozen=True, default=None)
+    status: LabelingServiceStatus = Field(frozen=True,
+                                          default=LabelingServiceStatus.Missing)
+    data_rows_count: int = Field(frozen=True)
+    data_rows_in_review_count: int = Field(frozen=True)
+    data_rows_in_rework_count: int = Field(frozen=True)
+    data_rows_done_count: int = Field(frozen=True)
 
     client: Any  # type Any to avoid circular import from client
 
@@ -46,6 +64,14 @@ def __init__(self, **kwargs):
             raise RuntimeError(
                 "Please enable experimental in client to use LabelingService")
 
+    @property
+    def tasks_completed(self):
+        return self.data_rows_done_count
+
+    @property
+    def tasks_remaining(self):
+        return self.data_rows_count - self.data_rows_done_count
+
     class Config(_CamelCaseMixin.Config):
         ...
 
@@ -124,13 +150,8 @@ def convert_to_labeling_service_dashboard(client, data):
         )
 
     @root_validator(pre=True)
-    def convert_graphql_to_attrs(cls, data):
+    def convert_boost_status_to_enum(cls, data):
         if 'boostStatus' in data:
             data['status'] = LabelingServiceStatus(data.pop('boostStatus'))
-        if 'dataRowsDoneCount' in data:
-            data['tasksCompleted'] = data.pop('dataRowsDoneCount')
-        if 'dataRowsCount' in data and 'dataRowsInReviewCount' in data and 'dataRowsInReworkCount' in data:
-            data['tasksRemaining'] = data['dataRowsCount'] - (
-                data['dataRowsInReviewCount'] + data['dataRowsInReworkCount'])
 
         return data