kanshan/ollama-python
1
0
Fork 0
mirror of https://github.com/ollama/ollama-python.git synced 2026-05-03 12:52:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Make ruff happy

Browse Source
This commit is contained in:
ParthSareen
2025-01-13 14:59:14 -08:00
parent 4dec73e8be
commit f718dab45d
3 changed files with 29 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_client.py
+2 -3
View File
@@ -536,7 +536,6 @@ def test_client_push_stream(httpserver: HTTPServer):
assert part['status'] == next(it)
@pytest.fixture
def userhomedir():
with tempfile.TemporaryDirectory() as temp:
@@ -559,7 +558,7 @@ def test_client_create_with_blob(httpserver: HTTPServer):
client = Client(httpserver.url_for('/'))
with tempfile.NamedTemporaryFile() as blob:
with tempfile.NamedTemporaryFile():
response = client.create('dummy', files={'test.gguf': 'sha256:e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855'})
assert response['status'] == 'success'
@@ -584,7 +583,7 @@ def test_client_create_with_parameters_roundtrip(httpserver: HTTPServer):
client = Client(httpserver.url_for('/'))
with tempfile.NamedTemporaryFile() as blob:
with tempfile.NamedTemporaryFile():
response = client.create(
'dummy',
quantize='q4_k_m',