Fix postcommit python dependency workflow (#33865)

akashorabek · web-flow · commit ccfc648b01c5 · 2025-02-05T10:41:06.000-05:00
diff --git a/sdks/python/apache_beam/ml/transforms/embeddings/huggingface_test.py b/sdks/python/apache_beam/ml/transforms/embeddings/huggingface_test.py
@@ -328,7 +328,7 @@ def test_sentence_transformer_images_with_str_data_types(self):
                     embedding_config))
 
 
-@unittest.skipIf(_HF_TOKEN is None, 'HF_TOKEN environment variable not set.')
+@unittest.skipIf(not _HF_TOKEN, 'HF_TOKEN environment variable not set.')
 class HuggingfaceInferenceAPITest(unittest.TestCase):
   def setUp(self):
     self.artifact_location = tempfile.mkdtemp()
@@ -366,7 +366,7 @@ def test_embeddings_with_inference_api(self):
       assert_that(max_ele_pcoll, equal_to(expected_output))
 
 
-@unittest.skipIf(_HF_TOKEN is None, 'HF_TOKEN environment variable not set.')
+@unittest.skipIf(not _HF_TOKEN, 'HF_TOKEN environment variable not set.')
 class HuggingfaceInferenceAPIGCSLocationTest(HuggingfaceInferenceAPITest):
   def setUp(self):
     self.artifact_location = self.gcs_artifact_location = os.path.join(
diff --git a/sdks/python/tox.ini b/sdks/python/tox.ini
@@ -463,7 +463,7 @@ deps =
   447: transformers>=4.47.0,<4.48.0
   447: torch>=1.9.0,<1.14.0
   448: transformers>=4.48.0,<4.49.0
-  448: torch>=2.0.0,torch<2.1.0
+  448: torch>=2.0.0,<2.1.0
   latest: transformers>=4.48.0
   latest: torch>=2.0.0
   tensorflow==2.12.0
