benchmarks: fix for local versions

iterative · Aug 20, 2024 · e76aaea · e76aaea
1 parent 6f24d04
commit e76aaea
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/dvc/testing/benchmarks/fixtures.py b/dvc/testing/benchmarks/fixtures.py
@@ -126,15 +126,17 @@ def _dvc_bin(*args):
 def dvc_bin(request, make_dvc_bin):
     if marker := request.node.get_closest_marker("requires"):
         from packaging.specifiers import SpecifierSet
-        from packaging.version import Version
+        from packaging.version import Version, parse
 
         spec = first(marker.args)
         assert spec is not None
         spec = SpecifierSet(spec) if isinstance(spec, str) else spec
         reason = marker.kwargs["reason"]
-        if Version(make_dvc_bin.version) not in spec:
+        dvc_version = make_dvc_bin.version
+        version = Version(parse(dvc_version).base_version)
+        if version not in spec:
             pytest.skip(
-                f"Version {make_dvc_bin.version} "
+                f"Version {dvc_version} "
                 f"does not satisfy requirement {spec!r}: {reason}"
             )
     return make_dvc_bin