From e76aaeac57620dc08e19e5c285d7d25261a99a93 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Saugat=20Pachhai=20=28=E0=A4=B8=E0=A5=8C=E0=A4=97=E0=A4=BE?= =?UTF-8?q?=E0=A4=A4=29?= Date: Tue, 20 Aug 2024 21:32:37 +0545 Subject: [PATCH] benchmarks: fix for local versions --- dvc/testing/benchmarks/fixtures.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/dvc/testing/benchmarks/fixtures.py b/dvc/testing/benchmarks/fixtures.py index 15d2da5b9..e59048ba3 100644 --- a/dvc/testing/benchmarks/fixtures.py +++ b/dvc/testing/benchmarks/fixtures.py @@ -126,15 +126,17 @@ def make_dvc_bin( 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