vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix coding style

Browse source
This commit is contained in:
Xuan Son Nguyen 2024-11-20 17:58:14 +01:00
parent 472e128c0b
commit 6af3f95f6f
3 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

2
examples/server/tests/unit/test_rerank.py
View file

@ -24,7 +24,7 @@ def test_rerank():
}) })
assert res.status_code == 200 assert res.status_code == 200
assert len(res.body["results"]) == 4 assert len(res.body["results"]) == 4
most_relevant = res.body["results"][0] most_relevant = res.body["results"][0]
least_relevant = res.body["results"][0] least_relevant = res.body["results"][0]
for doc in res.body["results"]: for doc in res.body["results"]:

2
examples/server/tests/unit/test_slot_save.py
View file

@ -47,7 +47,7 @@ def test_slot_save_restore():
}) })
assert res.status_code == 200 assert res.status_code == 200
assert res.body["n_restored"] == 84 assert res.body["n_restored"] == 84
# Since we have cache, slot 0 should only process the last tokens # Since we have cache, slot 0 should only process the last tokens
res = server.make_request("POST", "/completion", data={ res = server.make_request("POST", "/completion", data={
"prompt": "What is the capital of Germany?", "prompt": "What is the capital of Germany?",

4
examples/server/tests/utils.py
View file

@ -292,7 +292,7 @@ class ServerPreset:
server.temperature = 0.0 server.temperature = 0.0
server.seed = 42 server.seed = 42
return server return server
@staticmethod @staticmethod
def stories15m_moe() -> ServerProcess: def stories15m_moe() -> ServerProcess:
server = ServerProcess() server = ServerProcess()
@ -306,7 +306,7 @@ class ServerPreset:
server.temperature = 0.0 server.temperature = 0.0
server.seed = 42 server.seed = 42
return server return server
@staticmethod @staticmethod
def jina_reranker_tiny() -> ServerProcess: def jina_reranker_tiny() -> ServerProcess:
server = ServerProcess() server = ServerProcess()