Merge branch 'lean/yolov12-upload' of github.com:roboflow/roboflow-python into lean/yolov12-upload

Author: lrosemberg

roboflow/util/model_processor.py

@@ -56,7 +56,7 @@ def _get_processor_function(model_type: str) -> Callable:
 
     if "yolonas" in model_type:
         return _process_yolonas
-    
+
     if "yolov12" in model_type:
         return _process_yolov12
 
@@ -123,11 +123,7 @@ def _process_yolo(model_type: str, model_path: str, filename: str) -> str:
 
     if "yolov8" in model_type or "yolov10" in model_type or "yolov11" in model_type:
         # try except for backwards compatibility with older versions of ultralytics
-        if (
-            "-cls" in model_type
-            or model_type.startswith("yolov10")
-            or model_type.startswith("yolov11")
-        ):
+        if "-cls" in model_type or model_type.startswith("yolov10") or model_type.startswith("yolov11"):
             nc = model["model"].yaml["nc"]
             args = model["train_args"]
         else:
@@ -215,44 +211,27 @@ def _process_yolov12(model_type: str, model_path: str, filename: str) -> str:
 
     # Find any .pt file in model path
     model_files = os.listdir(model_path)
-    pt_file = next((f for f in model_files if f.endswith('.pt')), None)
-    
+    pt_file = next((f for f in model_files if f.endswith(".pt")), None)
+
     if pt_file is None:
         raise RuntimeError("No .pt model file found in the provided path")
 
     # Copy the .pt file to weights.pt if not already named weights.pt
     if pt_file != "weights.pt":
-        shutil.copy(
-            os.path.join(model_path, pt_file),
-            os.path.join(model_path, "weights.pt")
-        )
+        shutil.copy(os.path.join(model_path, pt_file), os.path.join(model_path, "weights.pt"))
 
-    required_files = [
-        "weights.pt"
-    ]
+    required_files = ["weights.pt"]
 
-    optional_files = [
-        "results.csv", 
-        "results.png",
-        "model_artifacts.json"
-    ]
+    optional_files = ["results.csv", "results.png", "model_artifacts.json"]
 
     zip_file_name = "roboflow_deploy.zip"
     with zipfile.ZipFile(os.path.join(model_path, zip_file_name), "w") as zipMe:
         for file in required_files:
-            zipMe.write(
-                os.path.join(model_path, file),
-                arcname=file,
-                compress_type=zipfile.ZIP_DEFLATED
-            )
-        
+            zipMe.write(os.path.join(model_path, file), arcname=file, compress_type=zipfile.ZIP_DEFLATED)
+
         for file in optional_files:
             if os.path.exists(os.path.join(model_path, file)):
-                zipMe.write(
-                    os.path.join(model_path, file),
-                    arcname=file,
-                    compress_type=zipfile.ZIP_DEFLATED
-                )
+                zipMe.write(os.path.join(model_path, file), arcname=file, compress_type=zipfile.ZIP_DEFLATED)
 
     return zip_file_name