🛠️ fix dataset download for models other than YOLOv8

Author: SkalskiP

roboflow/core/version.py

@@ -570,7 +570,7 @@ def callback(content: dict) -> dict:
             if format in ["yolov5pytorch", "yolov7pytorch", "yolov8"]:
                 content["train"] = location + content["train"].lstrip("..")
                 content["val"] = location + content["val"].lstrip("..")
-            if not get_wrong_dependencies_versions([("ultralytics", ">=", "8.0.30")]):
+            if not get_wrong_dependencies_versions([("ultralytics", ">=", "8.0.30")], pass_uninstalled=True):
                 content["train"] = "train/images"
                 content["val"] = "valid/images"
                 content["test"] = "test/images"

roboflow/util/versions.py

@@ -5,16 +5,17 @@
 
 
 def get_wrong_dependencies_versions(
-    dependencies_versions: List[Tuple[str, str, str]]
+    dependencies_versions: List[Tuple[str, str, str]], pass_uninstalled: bool = False
 ) -> List[Tuple[str, str, str, str]]:
     """
-    Get a list of missmatching dependencies with current version installed.
+    Get a list of mismatching dependencies with current version installed.
     E.g., assuming we pass `get_wrong_dependencies_versions([("torch", "==", "1.2.0")]), we will check if the current version of `torch` is `==1.2.0`. If not, we will return `[("torch", "==", "1.2.0", "<current_installed_version>")]
 
     We support `<=`, `==`, `>=`
 
     Args:
         dependencies_versions (List[Tuple[str, str]]): List of dependencies we want to check, [("<package_name>", "<version_number_to_check")]
+        pass_uninstalled (bool): By default get_wrong_dependencies_versions will throw exception if <package_name> is not installed. You can pass_uninstalled packages by setting this parameter to True.
 
     Returns:
         List[Tuple[str, str, str]]: List of dependencies with wrong version, [("<package_name>", "<version_number_to_check", "<current_version>")]
@@ -26,18 +27,23 @@ def get_wrong_dependencies_versions(
         "<=": lambda x, y: x <= y,
     }
     for dependency, order, version in dependencies_versions:
-        module = import_module(dependency)
-        module_version = module.__version__
-        if order not in order_funcs:
-            raise ValueError(
-                f"order={order} not supported, please use `{', '.join(order_funcs.keys())}`"
-            )
-
-        is_okay = order_funcs[order](Version(module_version), Version(version))
-        if not is_okay:
-            wrong_dependencies_versions.append(
-                (dependency, order, version, module_version)
-            )
+        try:
+            module = import_module(dependency)
+            module_version = module.__version__
+            if order not in order_funcs:
+                raise ValueError(
+                    f"order={order} not supported, please use `{', '.join(order_funcs.keys())}`"
+                )
+
+            is_okay = order_funcs[order](Version(module_version), Version(version))
+            if not is_okay:
+                wrong_dependencies_versions.append(
+                    (dependency, order, version, module_version)
+                )
+        except Exception as e:
+            if pass_uninstalled:
+                continue
+            raise e
     return wrong_dependencies_versions