This is an automated email from the ASF dual-hosted git repository.
yhu pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/beam.git
The following commit(s) were added to refs/heads/master by this push:
new c160a081923 Fix Tensorflow intergration test model path (#25553)
c160a081923 is described below
commit c160a081923c45d8c9304a57802533b744a54663
Author: Anand Inguva <[email protected]>
AuthorDate: Sun Feb 19 11:51:51 2023 -0500
Fix Tensorflow intergration test model path (#25553)
* Fix model path
* Fix tox
---
.../python/apache_beam/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 7b4b13ce2e1..3c92461c15a 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 @@ class TensorflowInference(unittest.TestCase):
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 5b7e10bf12a..c21e384ca86 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
