style: apply ruff format to all files

mlflow_utils/model_registry.py

@@ -192,7 +192,7 @@ def register_model_for_kserve(
             tags={
                 "model.type": model_type,
                 "deployment.target": "kserve",
-            }
+            },
         )
         logger.info(f"Created registered model: {model_name}")
 
@@ -206,13 +206,10 @@ def register_model_for_kserve(
             **(tags or {}),
             "model.type": model_type,
             **kserve_config.as_dict(),
-        }
+        },
     )
 
-    logger.info(
-        f"Registered model version {model_version.version} "
-        f"for {model_name} (type: {model_type})"
-    )
+    logger.info(f"Registered model version {model_version.version} for {model_name} (type: {model_type})")
 
     return model_version
 
@@ -412,25 +409,24 @@ def generate_kserve_manifest(
 
     if config.container_image:
         # Custom container
-        predictor["containers"] = [{
-            "name": "predictor",
-            "image": config.container_image,
-            "ports": [{"containerPort": config.container_port, "protocol": "TCP"}],
-            "resources": {
-                "requests": {
-                    "cpu": config.cpu_request,
-                    "memory": config.memory_request,
+        predictor["containers"] = [
+            {
+                "name": "predictor",
+                "image": config.container_image,
+                "ports": [{"containerPort": config.container_port, "protocol": "TCP"}],
+                "resources": {
+                    "requests": {
+                        "cpu": config.cpu_request,
+                        "memory": config.memory_request,
+                    },
+                    "limits": {
+                        "cpu": config.cpu_limit,
+                        "memory": config.memory_limit,
+                    },
                 },
-                "limits": {
-                    "cpu": config.cpu_limit,
-                    "memory": config.memory_limit,
-                },
-            },
-            "env": [
-                {"name": k, "value": v}
-                for k, v in config.env_vars.items()
-            ],
-        }]
+                "env": [{"name": k, "value": v} for k, v in config.env_vars.items()],
+            }
+        ]
 
         # Add GPU if needed
         if config.gpu_count > 0:
@@ -494,7 +490,7 @@ def generate_kserve_yaml(
     yaml_str = yaml.dump(manifest, default_flow_style=False, sort_keys=False)
 
     if output_path:
-        with open(output_path, 'w') as f:
+        with open(output_path, "w") as f:
             f.write(yaml_str)
         logger.info(f"Wrote KServe manifest to {output_path}")