linting

apache · Jun 11, 2024 · d58104d · d58104d
1 parent 7d3d242
commit d58104d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/sdks/python/apache_beam/ml/transforms/embeddings/tensorflow_hub_test.py b/sdks/python/apache_beam/ml/transforms/embeddings/tensorflow_hub_test.py
@@ -33,7 +33,6 @@
 # pylint: disable=ungrouped-imports
 try:
   from apache_beam.ml.transforms.embeddings.tensorflow_hub import TensorflowHubTextEmbeddings
-  import tensorflow as tf
 except ImportError:
   TensorflowHubTextEmbeddings = None  # type: ignore
   tf = None  # type: ignore
@@ -49,7 +48,7 @@
 try:
   from apache_beam.ml.transforms.embeddings.tensorflow_hub import TensorflowHubImageEmbeddings
   from PIL import Image
-except:
+except ImportError:
   TensorflowHubImageEmbeddings = None  # type: ignore
   Image = None  # type: ignore