fix lint

Author: TarunRavikumar

model-engine/model_engine_server/common/io.py

@@ -34,9 +34,7 @@ def open_wrapper(uri: str, mode: str = "rt", **kwargs):
             client_kwargs["endpoint_url"] = s3_endpoint
 
         addressing_style = getattr(infra_config(), "s3_addressing_style", "path")
-        client_kwargs["config"] = boto3.session.Config(
-            s3={"addressing_style": addressing_style}
-        )
+        client_kwargs["config"] = boto3.session.Config(s3={"addressing_style": addressing_style})
 
         client = session.client("s3", **client_kwargs)
     else:

model-engine/model_engine_server/core/celery/app.py

@@ -544,7 +544,9 @@ def _get_backend_url_and_conf(
                 }
             )
         else:
-            logger.info("Non-AWS deployment, using environment variables for S3 backend credentials")
+            logger.info(
+                "Non-AWS deployment, using environment variables for S3 backend credentials"
+            )
             out_conf_changes.update(
                 {
                     "s3_bucket": s3_bucket,

model-engine/model_engine_server/domain/entities/model_bundle_entity.py

@@ -77,6 +77,7 @@ def validate_fields_present_for_framework_type(cls, field_values):
             )
             if not field_values.get("ecr_repo"):
                 from model_engine_server.core.config import infra_config
+
                 if infra_config().cloud_provider != "onprem":
                     raise ValueError(
                         "Expected `ecr_repo` to be non-null for custom framework. "

model-engine/model_engine_server/infra/gateways/resources/onprem_queue_endpoint_resource_delegate.py

@@ -30,9 +30,7 @@ async def create_queue_if_not_exists(
 
     async def delete_queue(self, endpoint_id: str) -> None:
         queue_name = QueueEndpointResourceDelegate.endpoint_id_to_queue_name(endpoint_id)
-        logger.debug(
-            f"Delete request for queue {queue_name} (no-op for Redis-based queues)"
-        )
+        logger.debug(f"Delete request for queue {queue_name} (no-op for Redis-based queues)")
 
     async def get_queue_attributes(self, endpoint_id: str) -> Dict[str, Any]:
         queue_name = QueueEndpointResourceDelegate.endpoint_id_to_queue_name(endpoint_id)

model-engine/model_engine_server/infra/gateways/s3_file_storage_gateway.py

@@ -93,7 +93,7 @@ async def list_files(self, owner: str) -> List[FileMetadata]:
             for obj in objects.get("Contents", []):
                 key = obj["Key"]
                 if key.startswith(owner):
-                    file_id = key[len(owner):].lstrip("/")
+                    file_id = key[len(owner) :].lstrip("/")
                     if file_id:
                         file_metadata = await self.get_file(owner, file_id)
                         if file_metadata:

model-engine/model_engine_server/infra/gateways/s3_llm_artifact_gateway.py

@@ -23,9 +23,7 @@ def list_files(self, path: str, **kwargs) -> List[str]:
         logger.debug(f"Listed {len(files)} files from {path}")
         return files
 
-    def download_files(
-        self, path: str, target_path: str, overwrite=False, **kwargs
-    ) -> List[str]:
+    def download_files(self, path: str, target_path: str, overwrite=False, **kwargs) -> List[str]:
         s3 = get_s3_resource(kwargs)
         parsed_remote = parse_attachment_url(path, clean_key=False)
         bucket = parsed_remote.bucket

model-engine/model_engine_server/infra/gateways/s3_utils.py

@@ -12,54 +12,49 @@ def get_s3_client(kwargs: Optional[Dict[str, Any]] = None):
     kwargs = kwargs or {}
     session = boto3.Session()
     client_kwargs = {}
-    
+
     if infra_config().cloud_provider == "onprem":
         logger.debug("Using on-prem/MinIO S3-compatible configuration")
-        
+
         s3_endpoint = getattr(infra_config(), "s3_endpoint_url", None) or os.getenv(
             "S3_ENDPOINT_URL"
         )
         if s3_endpoint:
             client_kwargs["endpoint_url"] = s3_endpoint
             logger.debug(f"Using S3 endpoint: {s3_endpoint}")
-        
+
         addressing_style = getattr(infra_config(), "s3_addressing_style", "path")
-        client_kwargs["config"] = boto3.session.Config(
-            s3={"addressing_style": addressing_style}
-        )
+        client_kwargs["config"] = boto3.session.Config(s3={"addressing_style": addressing_style})
     else:
         logger.debug("Using AWS S3 configuration")
         aws_profile = kwargs.get("aws_profile")
         if aws_profile:
             session = boto3.Session(profile_name=aws_profile)
-    
+
     return session.client("s3", **client_kwargs)
 
 
 def get_s3_resource(kwargs: Optional[Dict[str, Any]] = None):
     kwargs = kwargs or {}
     session = boto3.Session()
     resource_kwargs = {}
-    
+
     if infra_config().cloud_provider == "onprem":
         logger.debug("Using on-prem/MinIO S3-compatible configuration")
-        
+
         s3_endpoint = getattr(infra_config(), "s3_endpoint_url", None) or os.getenv(
             "S3_ENDPOINT_URL"
         )
         if s3_endpoint:
             resource_kwargs["endpoint_url"] = s3_endpoint
             logger.debug(f"Using S3 endpoint: {s3_endpoint}")
-        
+
         addressing_style = getattr(infra_config(), "s3_addressing_style", "path")
-        resource_kwargs["config"] = boto3.session.Config(
-            s3={"addressing_style": addressing_style}
-        )
+        resource_kwargs["config"] = boto3.session.Config(s3={"addressing_style": addressing_style})
     else:
         logger.debug("Using AWS S3 configuration")
         aws_profile = kwargs.get("aws_profile")
         if aws_profile:
             session = boto3.Session(profile_name=aws_profile)
-    
-    return session.resource("s3", **resource_kwargs)
 
+    return session.resource("s3", **resource_kwargs)

model-engine/model_engine_server/infra/repositories/s3_file_llm_fine_tune_repository.py

@@ -22,7 +22,7 @@ def _open(self, uri: str, mode: str = "rt", **kwargs) -> IO:
 
     @staticmethod
     def _get_key(model_name: str, fine_tuning_method: str) -> str:
-        return f"{model_name}-{fine_tuning_method}" # possible for collisions but we control these names
+        return f"{model_name}-{fine_tuning_method}"  # possible for collisions but we control these names
 
     async def get_job_template_for_model(
         self, model_name: str, fine_tuning_method: str

model-engine/model_engine_server/infra/services/live_endpoint_builder_service.py

@@ -251,8 +251,7 @@ async def build_endpoint(
                     flavor = model_bundle.flavor
                     assert isinstance(flavor, RunnableImageLike)
                     image = self.docker_repository.get_image_url(
-                        image_tag=flavor.tag,
-                        repository_name=flavor.repository
+                        image_tag=flavor.tag, repository_name=flavor.repository
                     )
 
                 # Because this update is not the final update in the lock, the 'update_in_progress'