fix coding style
This commit is contained in:
parent
472e128c0b
commit
6af3f95f6f
3 changed files with 4 additions and 4 deletions
|
@ -24,7 +24,7 @@ def test_rerank():
|
|||
})
|
||||
assert res.status_code == 200
|
||||
assert len(res.body["results"]) == 4
|
||||
|
||||
|
||||
most_relevant = res.body["results"][0]
|
||||
least_relevant = res.body["results"][0]
|
||||
for doc in res.body["results"]:
|
||||
|
|
|
@ -47,7 +47,7 @@ def test_slot_save_restore():
|
|||
})
|
||||
assert res.status_code == 200
|
||||
assert res.body["n_restored"] == 84
|
||||
|
||||
|
||||
# Since we have cache, slot 0 should only process the last tokens
|
||||
res = server.make_request("POST", "/completion", data={
|
||||
"prompt": "What is the capital of Germany?",
|
||||
|
|
|
@ -292,7 +292,7 @@ class ServerPreset:
|
|||
server.temperature = 0.0
|
||||
server.seed = 42
|
||||
return server
|
||||
|
||||
|
||||
@staticmethod
|
||||
def stories15m_moe() -> ServerProcess:
|
||||
server = ServerProcess()
|
||||
|
@ -306,7 +306,7 @@ class ServerPreset:
|
|||
server.temperature = 0.0
|
||||
server.seed = 42
|
||||
return server
|
||||
|
||||
|
||||
@staticmethod
|
||||
def jina_reranker_tiny() -> ServerProcess:
|
||||
server = ServerProcess()
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue