Fix types

Author: midhun-pm

libs/labelbox/src/labelbox/alignerr/alignerr_project.py

@@ -42,7 +42,7 @@ def __init__(
         self.project = project
 
     @property
-    def project(self) -> Optional["Project"]:
+    def project(self) -> "Project":
         return self._project
 
     @project.setter
@@ -124,7 +124,7 @@ def add_tag(self, tag: EnhancedResourceTag):
 
         if tag.text not in current_tag_names:
             current_tag_names.append(tag.text)
-            self.set_tags(current_tag_names)
+            self.set_tags(current_tag_names, tag.type)
 
         return self
 
@@ -139,7 +139,7 @@ def remove_tag(self, tag: EnhancedResourceTag):
         """
         current_tags = self.get_tags()
         current_tag_names = [t.text for t in current_tags if t.uid != tag.uid]
-        self.set_tags(current_tag_names)
+        self.set_tags(current_tag_names, tag.type)
         return self
 
     def get_project_owner(self) -> Optional[ProjectBoostWorkforce]:

libs/labelbox/src/labelbox/alignerr/alignerr_project_builder.py

@@ -35,7 +35,7 @@ class AlignerrProjectBuilder:
     def __init__(self, client: "Client"):
         self.client = client
         self._alignerr_rates: dict[str, ProjectRateInput] = {}
-        self._customer_rate: ProjectRateInput = None
+        self._customer_rate: Optional[ProjectRateInput] = None
         self._domains: list[ProjectDomain] = []
         self._enhanced_resource_tags: list[EnhancedResourceTag] = []
         self._project_owner_email: Optional[str] = None
@@ -62,7 +62,7 @@ def set_alignerr_role_rate(
             raise ValueError(f"Role {role_name.value} not found")
 
         role_id = self.role_name_to_id[role_name.value]
-        role_name = role_name.value
+        role_name_str = role_name.value
 
         # Convert datetime objects to ISO format strings
         effective_since_str = (
@@ -76,7 +76,7 @@ def set_alignerr_role_rate(
             else effective_until
         )
 
-        self._alignerr_rates[role_name] = ProjectRateInput(
+        self._alignerr_rates[role_name_str] = ProjectRateInput(
             rateForId=role_id,
             isBillRate=False,
             billingMode=billing_mode,
@@ -221,7 +221,7 @@ def _create_resource_tags(self, alignerr_project: "AlignerrProject"):
                 f"Setting enhanced resource tags: {[tag.text for tag in self._enhanced_resource_tags]}"
             )
             # Group tags by type and set them accordingly
-            tags_by_type = {}
+            tags_by_type: dict[ResourceTagType, list[str]] = {}
             for tag in self._enhanced_resource_tags:
                 tag_type = tag.type
                 if tag_type not in tags_by_type:

libs/labelbox/src/labelbox/schema/role.py

@@ -30,7 +30,7 @@ class Role(DbObject):
     name = Field.String("name")
 
     @classmethod
-    def from_name(cls, client: "Client", name: str) -> "Role":
+    def from_name(cls, client: "Client", name: str) -> Optional["Role"]:
         roles = get_roles(client)
         return roles.get(name.upper())