Skip to content

Commit af975cf

Browse files
fix(pre_commit): 🎨 auto format pre-commit hooks
1 parent 72f787d commit af975cf

File tree

1 file changed

+13
-3
lines changed

1 file changed

+13
-3
lines changed

roboflow/core/version.py

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -462,7 +462,17 @@ def deploy(self, model_type: str, model_path: str, filename: str = "weights/best
462462
if model_type.startswith("yolo11"):
463463
model_type = model_type.replace("yolo11", "yolov11")
464464

465-
supported_models = ["yolov5", "yolov7-seg", "yolov8", "yolov9", "yolonas", "paligemma", "yolov10", "florence-2", "yolov11"]
465+
supported_models = [
466+
"yolov5",
467+
"yolov7-seg",
468+
"yolov8",
469+
"yolov9",
470+
"yolonas",
471+
"paligemma",
472+
"yolov10",
473+
"florence-2",
474+
"yolov11",
475+
]
466476

467477
if not any(supported_model in model_type for supported_model in supported_models):
468478
raise (ValueError(f"Model type {model_type} not supported. Supported models are" f" {supported_models}"))
@@ -520,7 +530,7 @@ def deploy(self, model_type: str, model_path: str, filename: str = "weights/best
520530
"The torch python package is required to deploy yolov5 models."
521531
" Please install it with `pip install torch`"
522532
)
523-
533+
524534
elif "yolov11" in model_type:
525535
try:
526536
import torch
@@ -531,7 +541,7 @@ def deploy(self, model_type: str, model_path: str, filename: str = "weights/best
531541
"The ultralytics python package is required to deploy yolov10"
532542
" models. Please install it with `pip install ultralytics`"
533543
)
534-
544+
535545
print_warn_for_wrong_dependencies_versions([("ultralytics", ">=", "8.3.0")], ask_to_continue=True)
536546

537547
model = torch.load(os.path.join(model_path, filename))

0 commit comments

Comments
 (0)