lint

Author: midhun-pm

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

@@ -5,8 +5,13 @@
 
 from labelbox.alignerr.schema.project_rate import ProjectRateV2
 from labelbox.alignerr.schema.project_domain import ProjectDomain
-from labelbox.alignerr.schema.enchanced_resource_tags import EnhancedResourceTag, ResourceTagType
-from labelbox.alignerr.schema.project_boost_workforce import ProjectBoostWorkforce
+from labelbox.alignerr.schema.enchanced_resource_tags import (
+    EnhancedResourceTag,
+    ResourceTagType,
+)
+from labelbox.alignerr.schema.project_boost_workforce import (
+    ProjectBoostWorkforce,
+)
 from labelbox.pagination import PaginatedCollection
 
 logger = logging.getLogger(__name__)
@@ -73,17 +78,19 @@ def set_tags(self, tag_names: list[str], tag_type: ResourceTagType):
         tag_ids = []
         for tag_name in tag_names:
             # Search for the tag by text to get its ID
-            found_tags = EnhancedResourceTag.search_by_text(self.client, search_text=tag_name, tag_type=tag_type)
+            found_tags = EnhancedResourceTag.search_by_text(
+                self.client, search_text=tag_name, tag_type=tag_type
+            )
             if found_tags:
                 tag_ids.append(found_tags[0].id)
-        
+
         # Use the existing project resource tag functionality with IDs
         self.project.update_project_resource_tags(tag_ids)
         return self
 
     def get_tags(self) -> list[EnhancedResourceTag]:
         """Get enhanced resource tags associated with this project.
-        
+
         Returns:
             List of EnhancedResourceTag instances
         """
@@ -94,7 +101,9 @@ def get_tags(self) -> list[EnhancedResourceTag]:
             # Search for the corresponding EnhancedResourceTag by text (try different types)
             found_tags = []
             for tag_type in [ResourceTagType.Default, ResourceTagType.Billing]:
-                found_tags = EnhancedResourceTag.search_by_text(self.client, search_text=tag.text, tag_type=tag_type)
+                found_tags = EnhancedResourceTag.search_by_text(
+                    self.client, search_text=tag.text, tag_type=tag_type
+                )
                 if found_tags:
                     break
             if found_tags:
@@ -103,28 +112,28 @@ def get_tags(self) -> list[EnhancedResourceTag]:
 
     def add_tag(self, tag: EnhancedResourceTag):
         """Add a single enhanced resource tag to the project.
-        
+
         Args:
             tag: EnhancedResourceTag instance to add
-            
+
         Returns:
             Self for method chaining
         """
         current_tags = self.get_tags()
         current_tag_names = [t.text for t in current_tags]
-        
+
         if tag.text not in current_tag_names:
             current_tag_names.append(tag.text)
             self.set_tags(current_tag_names)
-        
+
         return self
 
     def remove_tag(self, tag: EnhancedResourceTag):
         """Remove a single enhanced resource tag from the project.
-        
+
         Args:
             tag: EnhancedResourceTag instance to remove
-            
+
         Returns:
             Self for method chaining
         """
@@ -135,13 +144,12 @@ def remove_tag(self, tag: EnhancedResourceTag):
 
     def get_project_owner(self) -> Optional[ProjectBoostWorkforce]:
         """Get the ProjectBoostWorkforce for this project.
-        
+
         Returns:
             ProjectBoostWorkforce instance or None if not found
         """
         return ProjectBoostWorkforce.get_by_project_id(
-            client=self.client, 
-            project_id=self.project.uid
+            client=self.client, project_id=self.project.uid
         )
 
 

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

@@ -6,15 +6,21 @@
 from labelbox.alignerr.schema.project_rate import BillingMode
 from labelbox.alignerr.schema.project_rate import ProjectRateInput
 from labelbox.alignerr.schema.project_domain import ProjectDomain
-from labelbox.alignerr.schema.enchanced_resource_tags import EnhancedResourceTag, ResourceTagType
-from labelbox.alignerr.schema.project_boost_workforce import ProjectBoostWorkforce
+from labelbox.alignerr.schema.enchanced_resource_tags import (
+    EnhancedResourceTag,
+    ResourceTagType,
+)
+from labelbox.alignerr.schema.project_boost_workforce import (
+    ProjectBoostWorkforce,
+)
 from labelbox.schema.media_type import MediaType
 
 logger = logging.getLogger(__name__)
 
 
 class ValidationType(Enum):
     """Enum for validation types that can be selectively skipped."""
+
     ALIGNERR_RATE = "AlignerrRate"
     CUSTOMER_RATE = "CustomerRate"
     PROJECT_OWNER = "ProjectOwner"
@@ -123,11 +129,11 @@ def set_domains(self, domains: list[str]):
 
     def set_tags(self, tag_texts: list[str], tag_type: ResourceTagType):
         """Set enhanced resource tags for the project.
-        
+
         Args:
             tag_texts: List of tag text values to search for and attach
             tag_type: Type filter for searching tags
-            
+
         Returns:
             Self for method chaining
         """
@@ -136,7 +142,7 @@ def set_tags(self, tag_texts: list[str], tag_type: ResourceTagType):
             existing_tags = EnhancedResourceTag.search_by_text(
                 self.client, search_text=tag_text, tag_type=tag_type
             )
-            
+
             if existing_tags:
                 # Use the first matching tag
                 self._enhanced_resource_tags.append(existing_tags[0])
@@ -146,25 +152,26 @@ def set_tags(self, tag_texts: list[str], tag_type: ResourceTagType):
                     self.client,
                     text=tag_text,
                     color="#007bff",  # Default blue color
-                    tag_type=tag_type
+                    tag_type=tag_type,
                 )
                 self._enhanced_resource_tags.append(new_tag)
         return self
 
     def set_project_owner(self, project_owner_email: str):
         """Set the project owner for the ProjectBoostWorkforce.
-        
+
         Args:
             project_owner_email: Email of the user to set as project owner
-            
+
         Returns:
             Self for method chaining
         """
         self._project_owner_email = project_owner_email
         return self
 
-
-    def create(self, skip_validation: Union[bool, List[ValidationType]] = False):
+    def create(
+        self, skip_validation: Union[bool, List[ValidationType]] = False
+    ):
         if not skip_validation:
             self._validate()
         elif isinstance(skip_validation, list):
@@ -220,7 +227,7 @@ def _create_resource_tags(self, alignerr_project: "AlignerrProject"):
                 if tag_type not in tags_by_type:
                     tags_by_type[tag_type] = []
                 tags_by_type[tag_type].append(tag.text)
-            
+
             # Set tags for each type
             for tag_type_str, tag_names in tags_by_type.items():
                 # Convert string back to enum
@@ -230,17 +237,19 @@ def _create_resource_tags(self, alignerr_project: "AlignerrProject"):
     def _create_project_owner(self, alignerr_project: "AlignerrProject"):
         if self._project_owner_email:
             logger.info(f"Setting project owner: {self._project_owner_email}")
-            
+
             # Find user by email in the organization
             user_id = self._find_user_by_email(self._project_owner_email)
             if not user_id:
                 current_org = self.client.get_organization()
-                raise ValueError(f"User with email {self._project_owner_email} not found in organization {current_org.uid}")
-            
+                raise ValueError(
+                    f"User with email {self._project_owner_email} not found in organization {current_org.uid}"
+                )
+
             ProjectBoostWorkforce.set_project_owner(
                 client=self.client,
                 project_id=alignerr_project.project.uid,
-                project_owner_user_id=user_id
+                project_owner_user_id=user_id,
             )
 
     def _validate_alignerr_rates(self):
@@ -273,16 +282,16 @@ def _validate(self):
 
     def _validate_selective(self, skip_validations: List[ValidationType]):
         """Run validations selectively, skipping those in the provided list.
-        
+
         Args:
             skip_validations: List of ValidationType enums to skip
         """
         if ValidationType.ALIGNERR_RATE not in skip_validations:
             self._validate_alignerr_rates()
-        
+
         if ValidationType.CUSTOMER_RATE not in skip_validations:
             self._validate_customer_rate()
-        
+
         if ValidationType.PROJECT_OWNER not in skip_validations:
             self._validate_project_owner()
 
@@ -292,31 +301,33 @@ def _get_role_name_to_id(self) -> dict[str, str]:
 
     def _find_user_by_email(self, email: str) -> Optional[str]:
         """Find user ID by email in the organization.
-        
+
         Args:
             email: Email address to search for
-            
+
         Returns:
             User ID if found, None otherwise
         """
         try:
             # Import here to avoid circular imports
             from labelbox.schema.user import User
-            
+
             # Get the current organization
             current_org = self.client.get_organization()
-            
+
             # Use client.get_users with where clause to find user by email
             users = self.client.get_users(where=User.email == email)
-            
+
             # Get the first matching user and verify they belong to the same organization
             user = next(users, None)
             if user and user.organization().uid == current_org.uid:
                 return user.uid
             else:
-                logger.warning(f"User with email {email} not found in organization {current_org.uid}")
+                logger.warning(
+                    f"User with email {email} not found in organization {current_org.uid}"
+                )
                 return None
-                    
+
         except Exception as e:
             logger.error(f"Error finding user by email {email}: {e}")
             return None

libs/labelbox/src/labelbox/alignerr/alignerr_project_factory.py

@@ -19,7 +19,9 @@ class AlignerrProjectFactory:
     def __init__(self, client: "Client"):
         self.client = client
 
-    def create(self, yaml_file_path: str, skip_validation: Union[bool, List] = False):
+    def create(
+        self, yaml_file_path: str, skip_validation: Union[bool, List] = False
+    ):
         """
         Create an AlignerrProject from a YAML configuration file.
 
@@ -236,46 +238,48 @@ def create(self, yaml_file_path: str, skip_validation: Union[bool, List] = False
             domains_config = config["domains"]
             if not isinstance(domains_config, list):
                 raise ValueError("'domains' must be a list")
-            
+
             if not all(isinstance(domain, str) for domain in domains_config):
                 raise ValueError("All domain names must be strings")
-            
+
             builder.set_domains(domains_config)
 
         # Set enhanced resource tags if provided
         if "tags" in config:
             tags_config = config["tags"]
             if not isinstance(tags_config, list):
                 raise ValueError("'tags' must be a list")
-            
+
             for tag_config in tags_config:
                 if not isinstance(tag_config, dict):
                     raise ValueError("Each tag must be a dictionary")
-                
+
                 required_tag_fields = ["text", "type"]
                 for field in required_tag_fields:
                     if field not in tag_config:
                         raise ValueError(
                             f"Required field '{field}' is missing for tag"
                         )
-                
+
                 # Validate tag type
                 try:
                     tag_type = ResourceTagType(tag_config["type"])
                 except ValueError:
                     raise ValueError(
                         f"Invalid tag type '{tag_config['type']}'. Must be one of: {[e.value for e in ResourceTagType]}"
                     )
-                
+
                 # Set the tag
                 builder.set_tags([tag_config["text"]], tag_type)
 
         # Set project owner if provided
         if "project_owner" in config:
             project_owner_config = config["project_owner"]
             if not isinstance(project_owner_config, str):
-                raise ValueError("'project_owner' must be a string (email address)")
-            
+                raise ValueError(
+                    "'project_owner' must be a string (email address)"
+                )
+
             builder.set_project_owner(project_owner_config)
 
         # Create the project