ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Skip the amd extra floor assertion while this branch has no amd extra

Browse source
This commit is contained in:
Daniel Han 2026-07-29 01:38:17 +00:00
commit 0d320c5b56
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
tests/python/test_cross_platform_parity.py
View file

@ -874,10 +874,15 @@ class TestAmdBnbFloorParity:
FLOOR = "0.50.0"
PYPROJECT = REPO_ROOT / "pyproject.toml"
def test_amd_extra_floor(self):
def test_amd_extra_floor_when_present(self):
"""This branch ships no `amd` extra: main's version carries direct URL
requirements, which PyPI rejects in Requires-Dist, so the extras block is not
merged here. Assert the floor once #7278 adds the extra in floor form, so it
cannot drift from the two installer fallbacks."""
text = self.PYPROJECT.read_text(encoding = "utf-8")
amd = re.search(r"^amd = \[(.*?)^\]", text, re.S | re.M)
assert amd, "pyproject.toml must define an `amd` extra"
if amd is None:
pytest.skip("pyproject.toml has no amd extra on this branch")
specs = re.findall(r'"(bitsandbytes[^"]*)"', amd.group(1))
assert specs, "the amd extra must pin bitsandbytes"
for spec in specs: