Skip to content

Commit

Permalink
Fix Tensorflow intergration test model path (#25553)
Browse files Browse the repository at this point in the history
* Fix model path

* Fix tox
  • Loading branch information
AnandInguva authored Feb 19, 2023
1 parent 977e531 commit c160a08
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down Expand Up @@ -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')
Expand Down
1 change: 0 additions & 1 deletion sdks/python/tox.ini
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit c160a08

Please sign in to comment.