From 02aa46d3d72213fa54650f4cbf860c3b86c07626 Mon Sep 17 00:00:00 2001 From: XQ Hu Date: Sun, 19 Jan 2025 19:14:40 +0000 Subject: [PATCH] pin sentence-transformers==3.3.1 --- sdks/python/test-suites/tox/py39/build.gradle | 4 +--- sdks/python/tox.ini | 2 +- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/sdks/python/test-suites/tox/py39/build.gradle b/sdks/python/test-suites/tox/py39/build.gradle index e9624f8e810e..ea02e9d5b1e8 100644 --- a/sdks/python/test-suites/tox/py39/build.gradle +++ b/sdks/python/test-suites/tox/py39/build.gradle @@ -168,9 +168,7 @@ postCommitPyDep.dependsOn "testPy39transformers-430" toxTask "testPy39embeddingsMLTransform", "py39-embeddings", "${posargs}" test.dependsOn "testPy39embeddingsMLTransform" -// TODO(https://github.com/apache/beam/issues/32965): re-enable this suite for the dep -// postcommit once the sentence-transformers import error is debugged -// postCommitPyDep.dependsOn "testPy39embeddingsMLTransform" +postCommitPyDep.dependsOn "testPy39embeddingsMLTransform" // Part of MLTransform embeddings test suite but requires tensorflow hub, which we need to test on // mutliple versions so keeping this suite separate. diff --git a/sdks/python/tox.ini b/sdks/python/tox.ini index ed1f723d6d4d..fe818324c194 100644 --- a/sdks/python/tox.ini +++ b/sdks/python/tox.ini @@ -481,7 +481,7 @@ commands = [testenv:py{39,310}-embeddings] deps = - sentence-transformers==2.2.2 + sentence-transformers==3.3.1 passenv = HF_INFERENCE_TOKEN extras = test,gcp commands =