You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@beam.apache.org by lo...@apache.org on 2023/02/20 00:57:31 UTC
[beam] 09/13: Fix Tensorflow intergration test model path (#25553)
This is an automated email from the ASF dual-hosted git repository.
lostluck pushed a commit to branch prism-jobservices
in repository https://gitbox.apache.org/repos/asf/beam.git
commit d7e879496495d3d74058eba6196e6d6bcccbfd60
Author: Anand Inguva <34...@users.noreply.github.com>
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