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

log less, fix embd test

Browse source
This commit is contained in:
Xuan Son Nguyen 2024-11-20 16:49:35 +01:00
parent e34c9d78a4
commit eb02373f76
3 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

4
.github/workflows/server.yml vendored
View file

@ -180,7 +180,7 @@ jobs:
run: | run: |
cd examples/server/tests cd examples/server/tests
$env:PYTHONIOENCODING = ":replace" $env:PYTHONIOENCODING = ":replace"
pytest -v -s -x pytest -v -x
- name: Slow tests - name: Slow tests
id: server_integration_tests_slow id: server_integration_tests_slow
@ -188,4 +188,4 @@ jobs:
run: | run: |
cd examples/server/tests cd examples/server/tests
$env:SLOW_TESTS = "1" $env:SLOW_TESTS = "1"
pytest -v -s -x pytest -v -x

2
examples/server/tests/tests.sh
View file

@ -4,7 +4,7 @@ set -eu
if [ $# -lt 1 ] if [ $# -lt 1 ]
then then
pytest -v -s -x pytest -v -x
else else
pytest "$@" pytest "$@"
fi fi

4
examples/server/tests/unit/test_embedding.py
View file

@ -23,7 +23,7 @@ def test_embedding_single():
assert len(res.body['data'][0]['embedding']) > 1 assert len(res.body['data'][0]['embedding']) > 1
# make sure embedding vector is normalized # make sure embedding vector is normalized
assert abs(sum([x ** 2 for x in res.body['data'][0]['embedding']]) - 1) < 1e-6 assert abs(sum([x ** 2 for x in res.body['data'][0]['embedding']]) - 1) < 1e-5
def test_embedding_multiple(): def test_embedding_multiple():
@ -95,4 +95,4 @@ def test_same_prompt_give_same_result():
v0 = res.body['data'][0]['embedding'] v0 = res.body['data'][0]['embedding']
vi = res.body['data'][i]['embedding'] vi = res.body['data'][i]['embedding']
for x, y in zip(v0, vi): for x, y in zip(v0, vi):
assert abs(x - y) < 1e-6 assert abs(x - y) < 1e-5