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
|
||||
not usable chat models."""
|
||||
from utils.models.hidden_models import is_hidden_model
|
||||
|
||||
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"]
|
||||
|
||||
|
||||
def test_cached_gguf_scan_keeps_infra_repo_with_user_downloaded_variant(
|
||||
monkeypatch, tmp_path
|
||||
):
|
||||
def test_cached_gguf_scan_keeps_infra_repo_with_user_downloaded_variant(monkeypatch, tmp_path):
|
||||
monkeypatch.setattr(state_dir, "cache_root", lambda: tmp_path / "state")
|
||||
embedder = _repo(
|
||||
"unsloth/bge-small-en-v1.5-GGUF",
|
||||
|
|
@ -469,11 +467,7 @@ def test_cached_gguf_scan_keeps_infra_repo_with_user_downloaded_variant(
|
|||
"model",
|
||||
"unsloth/bge-small-en-v1.5-GGUF",
|
||||
"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",
|
||||
)
|
||||
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()}
|
||||
|
||||
assert [row["repo_id"] for row in result["cached"]] == [
|
||||
"unsloth/bge-small-en-v1.5-GGUF"
|
||||
]
|
||||
assert [row["repo_id"] for row in result["cached"]] == ["unsloth/bge-small-en-v1.5-GGUF"]
|
||||
|
||||
|
||||
def test_cached_models_scan_hides_non_gguf_embedder(monkeypatch, tmp_path):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue