Skip to content

Commit 1c9c775

Browse files
fix(pre_commit): 🎨 auto format pre-commit hooks
1 parent 7adc7d7 commit 1c9c775

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

roboflow/util/model_processor.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -138,7 +138,12 @@ def _process_yolo(model_type: str, model_path: str, filename: str) -> str:
138138

139139
if "yolov8" in model_type or "yolov10" in model_type or "yolov11" in model_type or "yolov12" in model_type:
140140
# try except for backwards compatibility with older versions of ultralytics
141-
if "-cls" in model_type or model_type.startswith("yolov10") or model_type.startswith("yolov11") or model_type.startswith("yolov12"):
141+
if (
142+
"-cls" in model_type
143+
or model_type.startswith("yolov10")
144+
or model_type.startswith("yolov11")
145+
or model_type.startswith("yolov12")
146+
):
142147
nc = model["model"].yaml["nc"]
143148
args = model["train_args"]
144149
else:

0 commit comments

Comments
 (0)