From 43a4189c79d6aec83010a644f1dc4fb506c3193d Mon Sep 17 00:00:00 2001 From: Andrey Cheptsov Date: Fri, 27 Mar 2026 18:36:30 +0100 Subject: [PATCH 1/2] Fix integrity tests for CloudRift and Runpod drift --- pyproject.toml | 2 +- src/integrity_tests/test_cloudrift.py | 2 +- src/integrity_tests/test_runpod.py | 6 ++++-- 3 files changed, 6 insertions(+), 4 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index ff10980..4462315 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -9,7 +9,7 @@ authors = [ ] description = "A catalog of GPU pricing for different cloud providers" readme = "README.md" -requires-python = ">=3.9" +requires-python = ">=3.10" classifiers = [ "Programming Language :: Python :: 3", "License :: OSI Approved :: Mozilla Public License 2.0 (MPL 2.0)", diff --git a/src/integrity_tests/test_cloudrift.py b/src/integrity_tests/test_cloudrift.py index 2ae66da..65e7134 100644 --- a/src/integrity_tests/test_cloudrift.py +++ b/src/integrity_tests/test_cloudrift.py @@ -18,7 +18,7 @@ def select_row(rows, name: str) -> list[str]: def test_gpu_present(data_rows: list[dict]): - expected_gpus = [gpu for _, gpu in GPU_MAP] + expected_gpus = [gpu for _, gpu, _vendor in GPU_MAP] gpus = select_row(data_rows, "gpu_name") gpus = list(dict.fromkeys(gpus)) assert set(gpus).issubset( diff --git a/src/integrity_tests/test_runpod.py b/src/integrity_tests/test_runpod.py index a8f0040..da101e9 100644 --- a/src/integrity_tests/test_runpod.py +++ b/src/integrity_tests/test_runpod.py @@ -37,8 +37,10 @@ def test_locations(data_rows): "US", } locations = set(select_row(data_rows, "location")) - # Assert most are present. Some may be missing due to low availability - assert len(expected - locations) <= 3 + # Assert most are present. Some may be missing due to low availability. + # TODO: CA-MTL-2 looks absent in recent live Runpod snapshots. + # Re-evaluate this expectation later and tighten back to <= 3. + assert len(expected - locations) <= 4 def test_spot(data_rows): From 206cb50fc60c9a984a40126752548db8ca6e66f8 Mon Sep 17 00:00:00 2001 From: Andrey Cheptsov Date: Fri, 27 Mar 2026 18:42:08 +0100 Subject: [PATCH 2/2] Revert minimum Python version to 3.9 --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 4462315..ff10980 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -9,7 +9,7 @@ authors = [ ] description = "A catalog of GPU pricing for different cloud providers" readme = "README.md" -requires-python = ">=3.10" +requires-python = ">=3.9" classifiers = [ "Programming Language :: Python :: 3", "License :: OSI Approved :: Mozilla Public License 2.0 (MPL 2.0)",