diff --git a/ann/src/main/python/dataflow/faiss_index_bq_dataset.py b/ann/src/main/python/dataflow/faiss_index_bq_dataset.py
index dd45070db..3cc238abd 100644
--- a/ann/src/main/python/dataflow/faiss_index_bq_dataset.py
+++ b/ann/src/main/python/dataflow/faiss_index_bq_dataset.py
@@ -4,6 +4,8 @@ import os
 import pkgutil
 import sys
 from urllib.parse import urlsplit
+import torch as nn
+import tensorflow as tf
 
 import apache_beam as beam
 from apache_beam.options.pipeline_options import PipelineOptions