From 528f94b2ccc2b926ff37c57a954aad49068dca80 Mon Sep 17 00:00:00 2001
From: ii_ruatnec <104310191+SUNGMYEONGGI@users.noreply.github.com>
Date: Wed, 5 Apr 2023 18:41:02 +0900
Subject: [PATCH] Update faiss_index_bq_dataset.py

torch & tensorflow pull request
---
 ann/src/main/python/dataflow/faiss_index_bq_dataset.py | 2 ++
 1 file changed, 2 insertions(+)

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