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

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-07-26 15:33:43 +00:00
commit b47c55be75
1 changed files with 9 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/python/test_docker_llama_cuda_backend.py
View file

@ -96,12 +96,12 @@ def test_flashinfer_jit_cache_tracks_flashinfer(dockerfile: str):
# disagree, and that exception kills the vLLM EngineCore, which is what
# Unsloth's GRPO fast_inference path runs on. A literal pin drifts the moment
# vLLM bumps its flashinfer requirement, so the version has to be derived.
assert "flashinfer-jit-cache==${FI_VER}" in dockerfile, (
"flashinfer-jit-cache must be pinned to the resolved flashinfer-python version"
)
assert not re.search(r"flashinfer-jit-cache==[0-9]", dockerfile), (
"a literal flashinfer-jit-cache version will drift away from flashinfer-python"
)
assert (
"flashinfer-jit-cache==${FI_VER}" in dockerfile
), "flashinfer-jit-cache must be pinned to the resolved flashinfer-python version"
assert not re.search(
r"flashinfer-jit-cache==[0-9]", dockerfile
), "a literal flashinfer-jit-cache version will drift away from flashinfer-python"
assert "import flashinfer" in dockerfile, (
"the build must prove flashinfer imports, or a mismatch stays silent "
"until the first vLLM engine start"
@ -114,6 +114,6 @@ def test_cli_can_reach_the_studio_backend(dockerfile: str):
# requirement rather than an unsloth[huggingface] one, so the base venv has
# to ask for it explicitly or the whole CLI dies on ModuleNotFoundError.
assert '"structlog"' in dockerfile, "the base venv must install structlog for unsloth_cli"
assert "from studio.backend.core.export import ExportBackend" in dockerfile, (
"a build-time import guard must prove the CLI can reach the studio backend"
)
assert (
"from studio.backend.core.export import ExportBackend" in dockerfile
), "a build-time import guard must prove the CLI can reach the studio backend"