Skip to content

Commit a033482

Browse files
committed
try again without the try except
1 parent bc1f8b5 commit a033482

1 file changed

Lines changed: 1 addition & 5 deletions

File tree

sdks/python/apache_beam/ml/inference/huggingface_inference.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -32,11 +32,7 @@
3232
import torch
3333
from transformers import AutoModel
3434
from transformers import Pipeline
35-
36-
try:
37-
from transformers import TFAutoModel
38-
except ImportError:
39-
TFAutoModel = Any
35+
from transformers import TFAutoModel
4036
from transformers import pipeline
4137

4238
from apache_beam.ml.inference import utils

0 commit comments

Comments
 (0)