aladac/tensors
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix formatting issues

Browse Source 
Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
Adam Ladachowski
2026-02-15 19:57:55 +01:00
parent 24aaca2a48
commit fb27f8a9b0
2 changed files with 6 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tensors/hf.py
+1 -3
View File
@@ -115,9 +115,7 @@ def get_hf_model(
# Get safetensor files # Get safetensor files
siblings = getattr(model, "siblings", None) or [] siblings = getattr(model, "siblings", None) or []
safetensor_files = [ safetensor_files = [
{"rfilename": s.rfilename, "size": getattr(s, "size", None)} {"rfilename": s.rfilename, "size": getattr(s, "size", None)} for s in siblings if s.rfilename.endswith(".safetensors")
for s in siblings
if s.rfilename.endswith(".safetensors")
] ]
model_dict["_safetensor_files"] = safetensor_files model_dict["_safetensor_files"] = safetensor_files
tests/test_server.py
+5 -15
View File
@@ -541,9 +541,7 @@ class TestCivitAISearch:
"""Test search handles network errors.""" """Test search handles network errors."""
import httpx import httpx
respx_mock.get("https://civitai.com/api/v1/models").mock( respx_mock.get("https://civitai.com/api/v1/models").mock(side_effect=httpx.RequestError("Connection failed"))
side_effect=httpx.RequestError("Connection failed")
)
response = civitai_api.get("/api/civitai/search") response = civitai_api.get("/api/civitai/search")
assert response.status_code == 500 assert response.status_code == 500
@@ -594,9 +592,7 @@ class TestCivitAIGetModel:
"""Test get model handles network errors.""" """Test get model handles network errors."""
import httpx import httpx
respx_mock.get("https://civitai.com/api/v1/models/12345").mock( respx_mock.get("https://civitai.com/api/v1/models/12345").mock(side_effect=httpx.RequestError("Connection failed"))
side_effect=httpx.RequestError("Connection failed")
)
response = civitai_api.get("/api/civitai/model/12345") response = civitai_api.get("/api/civitai/model/12345")
assert response.status_code == 500 assert response.status_code == 500
@@ -1125,9 +1121,7 @@ class TestDownloadEndpoints:
) )
) )
response = download_api.post( response = download_api.post("/api/download", json={"version_id": 333, "output_dir": str(custom_dir)})
"/api/download", json={"version_id": 333, "output_dir": str(custom_dir)}
)
assert response.status_code == 200 assert response.status_code == 200
assert str(custom_dir) in response.json()["destination"] assert str(custom_dir) in response.json()["destination"]
@@ -1369,9 +1363,7 @@ class TestDownloadBackgroundTasks:
download_routes._active_downloads[download_id] = {"id": download_id, "status": "queued"} download_routes._active_downloads[download_id] = {"id": download_id, "status": "queued"}
# Mock download to return failure # Mock download to return failure
monkeypatch.setattr( monkeypatch.setattr(download_routes, "download_model_with_progress", lambda *args, **kwargs: False)
download_routes, "download_model_with_progress", lambda *args, **kwargs: False
)
dest_path = tmp_path / "model.safetensors" dest_path = tmp_path / "model.safetensors"
_do_download(12345, dest_path, None, download_id) _do_download(12345, dest_path, None, download_id)
@@ -1580,9 +1572,7 @@ class TestDbFileLookup:
# Mock CivitAI hash lookup to return a match # Mock CivitAI hash lookup to return a match
respx_mock.get(f"https://civitai.com/api/v1/model-versions/by-hash/{sha256.upper()}").mock( respx_mock.get(f"https://civitai.com/api/v1/model-versions/by-hash/{sha256.upper()}").mock(
return_value=respx.MockResponse( return_value=respx.MockResponse(200, json={"id": 12345, "modelId": 67890, "name": "Found Model"})
200, json={"id": 12345, "modelId": 67890, "name": "Found Model"}
)
) )
response = db_api.post("/api/db/link") response = db_api.post("/api/db/link")