From eb02373f76fcd7d42a2798986120412333cf41ad Mon Sep 17 00:00:00 2001 From: Xuan Son Nguyen Date: Wed, 20 Nov 2024 16:49:35 +0100 Subject: [PATCH] log less, fix embd test --- .github/workflows/server.yml | 4 ++-- examples/server/tests/tests.sh | 2 +- examples/server/tests/unit/test_embedding.py | 4 ++-- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/.github/workflows/server.yml b/.github/workflows/server.yml index 9e1538831..2e8e3348f 100644 --- a/.github/workflows/server.yml +++ b/.github/workflows/server.yml @@ -180,7 +180,7 @@ jobs: run: | cd examples/server/tests $env:PYTHONIOENCODING = ":replace" - pytest -v -s -x + pytest -v -x - name: Slow tests id: server_integration_tests_slow @@ -188,4 +188,4 @@ jobs: run: | cd examples/server/tests $env:SLOW_TESTS = "1" - pytest -v -s -x + pytest -v -x diff --git a/examples/server/tests/tests.sh b/examples/server/tests/tests.sh index 130b09f99..1e285dcda 100755 --- a/examples/server/tests/tests.sh +++ b/examples/server/tests/tests.sh @@ -4,7 +4,7 @@ set -eu if [ $# -lt 1 ] then - pytest -v -s -x + pytest -v -x else pytest "$@" fi diff --git a/examples/server/tests/unit/test_embedding.py b/examples/server/tests/unit/test_embedding.py index a82885532..f287dfe20 100644 --- a/examples/server/tests/unit/test_embedding.py +++ b/examples/server/tests/unit/test_embedding.py @@ -23,7 +23,7 @@ def test_embedding_single(): assert len(res.body['data'][0]['embedding']) > 1 # 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(): @@ -95,4 +95,4 @@ def test_same_prompt_give_same_result(): v0 = res.body['data'][0]['embedding'] vi = res.body['data'][i]['embedding'] for x, y in zip(v0, vi): - assert abs(x - y) < 1e-6 + assert abs(x - y) < 1e-5