mirror of
https://github.com/unslothai/unsloth.git
synced 2026-07-09 15:58:41 +00:00
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
9547d2b128
commit
4ac010b73a
2 changed files with 3 additions and 12 deletions
|
|
@ -209,7 +209,6 @@ def _is_hidden_infra_repo(*values: str | None) -> bool:
|
||||||
validation probe) that are cached as a side effect of Studio itself and are
|
validation probe) that are cached as a side effect of Studio itself and are
|
||||||
not usable chat models."""
|
not usable chat models."""
|
||||||
from utils.models.hidden_models import is_hidden_model
|
from utils.models.hidden_models import is_hidden_model
|
||||||
|
|
||||||
return is_hidden_model(*values)
|
return is_hidden_model(*values)
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -452,9 +452,7 @@ def test_cached_gguf_scan_hides_infra_repos_without_user_downloads(monkeypatch,
|
||||||
assert [row["repo_id"] for row in result["cached"]] == ["Org/Chat-GGUF"]
|
assert [row["repo_id"] for row in result["cached"]] == ["Org/Chat-GGUF"]
|
||||||
|
|
||||||
|
|
||||||
def test_cached_gguf_scan_keeps_infra_repo_with_user_downloaded_variant(
|
def test_cached_gguf_scan_keeps_infra_repo_with_user_downloaded_variant(monkeypatch, tmp_path):
|
||||||
monkeypatch, tmp_path
|
|
||||||
):
|
|
||||||
monkeypatch.setattr(state_dir, "cache_root", lambda: tmp_path / "state")
|
monkeypatch.setattr(state_dir, "cache_root", lambda: tmp_path / "state")
|
||||||
embedder = _repo(
|
embedder = _repo(
|
||||||
"unsloth/bge-small-en-v1.5-GGUF",
|
"unsloth/bge-small-en-v1.5-GGUF",
|
||||||
|
|
@ -469,11 +467,7 @@ def test_cached_gguf_scan_keeps_infra_repo_with_user_downloaded_variant(
|
||||||
"model",
|
"model",
|
||||||
"unsloth/bge-small-en-v1.5-GGUF",
|
"unsloth/bge-small-en-v1.5-GGUF",
|
||||||
"Q8_0",
|
"Q8_0",
|
||||||
[
|
[download_manifest.ExpectedFile(path = "bge-small-en-v1.5-Q8_0.gguf", size = 35_000_000)],
|
||||||
download_manifest.ExpectedFile(
|
|
||||||
path = "bge-small-en-v1.5-Q8_0.gguf", size = 35_000_000
|
|
||||||
)
|
|
||||||
],
|
|
||||||
"http",
|
"http",
|
||||||
)
|
)
|
||||||
monkeypatch.setattr(
|
monkeypatch.setattr(
|
||||||
|
|
@ -489,9 +483,7 @@ def test_cached_gguf_scan_keeps_infra_repo_with_user_downloaded_variant(
|
||||||
|
|
||||||
result = {"cached": cache_inventory._scan_cached_gguf()}
|
result = {"cached": cache_inventory._scan_cached_gguf()}
|
||||||
|
|
||||||
assert [row["repo_id"] for row in result["cached"]] == [
|
assert [row["repo_id"] for row in result["cached"]] == ["unsloth/bge-small-en-v1.5-GGUF"]
|
||||||
"unsloth/bge-small-en-v1.5-GGUF"
|
|
||||||
]
|
|
||||||
|
|
||||||
|
|
||||||
def test_cached_models_scan_hides_non_gguf_embedder(monkeypatch, tmp_path):
|
def test_cached_models_scan_hides_non_gguf_embedder(monkeypatch, tmp_path):
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue