Skip to content

Commit 7e6da07

Browse files
committed
dont check ultralytics dep for v5 upload
1 parent d3f1972 commit 7e6da07

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

roboflow/core/version.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -294,7 +294,7 @@ def train(self, speed=None, checkpoint=None) -> bool:
294294

295295
return True
296296

297-
@warn_for_wrong_dependencies_versions([("ultralytics", "<=", "8.0.20")])
297+
# @warn_for_wrong_dependencies_versions([("ultralytics", "<=", "8.0.20")])
298298
def deploy(self, model_type: str, model_path: str) -> None:
299299
"""Uploads provided weights file to Roboflow
300300
@@ -315,10 +315,16 @@ def deploy(self, model_type: str, model_path: str) -> None:
315315
try:
316316
import torch
317317
import ultralytics
318+
318319
except ImportError as e:
319320
raise (
320321
"The ultralytics python package is required to deploy yolov8 models. Please install it with `pip install ultralytics`"
321322
)
323+
324+
print_warn_for_wrong_dependencies_versions(
325+
[("ultralytics", "<=", "8.0.20")]
326+
)
327+
322328
elif model_type == "yolov5":
323329
try:
324330
import torch

0 commit comments

Comments
 (0)