From c160a081923c45d8c9304a57802533b744a54663 Mon Sep 17 00:00:00 2001
From: Anand Inguva <34158215+AnandInguva@users.noreply.github.com>
Date: Sun, 19 Feb 2023 11:51:51 -0500
Subject: [PATCH] Fix Tensorflow intergration test model path (#25553)

* Fix model path

* Fix tox
---
 .../ml/inference/tensorflow_inference_it_test.py            | 6 +++---
 sdks/python/tox.ini                                         | 1 -
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/sdks/python/apache_beam/ml/inference/tensorflow_inference_it_test.py b/sdks/python/apache_beam/ml/inference/tensorflow_inference_it_test.py
index 7b4b13ce2e1e..3c92461c15af 100644
--- a/sdks/python/apache_beam/ml/inference/tensorflow_inference_it_test.py
+++ b/sdks/python/apache_beam/ml/inference/tensorflow_inference_it_test.py
@@ -51,9 +51,9 @@ class TensorflowInference(unittest.TestCase):
   def test_tf_mnist_classification(self):
     test_pipeline = TestPipeline(is_integration_test=True)
     input_file = 'gs://apache-beam-ml/testing/inputs/it_mnist_data.csv'
-    output_file_dir = 'apache-beam-ml/testing/outputs'
+    output_file_dir = 'gs://apache-beam-ml/testing/outputs'
     output_file = '/'.join([output_file_dir, str(uuid.uuid4()), 'result.txt'])
-    model_path = 'apache-beam-ml/models/tensorflow/mnist/'
+    model_path = 'gs://apache-beam-ml/models/tensorflow/mnist/'
     extra_opts = {
         'input': input_file,
         'output': output_file,
@@ -85,7 +85,7 @@ def test_tf_imagenet_image_segmentation(self):
     image_dir = (
         'https://storage.googleapis.com/download.tensorflow.org/example_images/'
     )
-    output_file_dir = 'apache-beam-ml/testing/outputs'
+    output_file_dir = 'gs://apache-beam-ml/testing/outputs'
     output_file = '/'.join([output_file_dir, str(uuid.uuid4()), 'result.txt'])
     model_path = (
         'https://tfhub.dev/google/tf2-preview/mobilenet_v2/classification/4')
diff --git a/sdks/python/tox.ini b/sdks/python/tox.ini
index 5b7e10bf12ab..c21e384ca866 100644
--- a/sdks/python/tox.ini
+++ b/sdks/python/tox.ini
@@ -344,7 +344,6 @@ commands =
   # Run all ONNX unit tests
   pytest -o junit_suite_name={envname} --junitxml=pytest_{envname}.xml -n 6 -m uses_onnx {posargs}
   
-[testenv:py{37,38,39,310}-tf-{211}]
 [testenv:py{37,38,39,310}-tensorflow-{29,210,211}]
 deps =
   -r build-requirements.txt