diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index ecef0569d3b0..0ee93afad4d8 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 6bf1806b4f14..a2e2f7585100 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 31c406ab8484..8c51992056c5 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