From 9ec5a915feb97cc1e207b3a2a3858cb114c17bc0 Mon Sep 17 00:00:00 2001 From: Ettore Di Giacinto Date: Fri, 17 Jan 2025 21:40:30 +0100 Subject: [PATCH] Adapt tests Signed-off-by: Ettore Di Giacinto --- .github/workflows/test.yml | 3 +-- core/http/app_test.go | 2 +- tests/models_fixtures/grpc.yaml | 2 +- 3 files changed, 3 insertions(+), 4 deletions(-) diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index ecef0569..0ee93afa 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -100,8 +100,7 @@ jobs: # The python3-grpc-tools package in 22.04 is too old pip install --user grpcio-tools - sudo rm -rfv /usr/bin/conda || true - PATH=$PATH:/opt/conda/bin make -C backend/python/sentencetransformers + make -C backend/python/transformers # Pre-build piper before we start tests in order to have shared libraries in place make sources/go-piper && \ diff --git a/core/http/app_test.go b/core/http/app_test.go index 6bf1806b..a2e2f758 100644 --- a/core/http/app_test.go +++ b/core/http/app_test.go @@ -822,7 +822,7 @@ var _ = Describe("API test", func() { application, err := application.New( append(commonOpts, - config.WithExternalBackend("huggingface", os.Getenv("HUGGINGFACE_GRPC")), + config.WithExternalBackend("transformers", os.Getenv("HUGGINGFACE_GRPC")), config.WithContext(c), config.WithModelPath(modelPath), )...) diff --git a/tests/models_fixtures/grpc.yaml b/tests/models_fixtures/grpc.yaml index 31c406ab..8c519920 100644 --- a/tests/models_fixtures/grpc.yaml +++ b/tests/models_fixtures/grpc.yaml @@ -1,5 +1,5 @@ name: code-search-ada-code-001 -backend: huggingface +backend: sentencetransformers embeddings: true parameters: model: all-MiniLM-L6-v2 \ No newline at end of file