diff --git a/.github/workflows/interactive-tensorflow.yml b/.github/workflows/interactive-tensorflow.yml index 6145d1c0c20..41b636fa536 100644 --- a/.github/workflows/interactive-tensorflow.yml +++ b/.github/workflows/interactive-tensorflow.yml @@ -27,5 +27,5 @@ jobs: - name: Interactive API - tensorflow_mnist run: | python setup.py build_grpc - pip install tensorflow==2.8 + pip install tensorflow==2.11 python -m tests.github.interactive_api_director.experiments.tensorflow_mnist.run diff --git a/tests/github/interactive_api_director/experiments/tensorflow_mnist/envoy/sd_requirements.txt b/tests/github/interactive_api_director/experiments/tensorflow_mnist/envoy/sd_requirements.txt index cb65dfddc4b..f4c454633d6 100644 --- a/tests/github/interactive_api_director/experiments/tensorflow_mnist/envoy/sd_requirements.txt +++ b/tests/github/interactive_api_director/experiments/tensorflow_mnist/envoy/sd_requirements.txt @@ -1 +1 @@ -tensorflow==2.8.4 +tensorflow==2.11.1 diff --git a/tests/github/interactive_api_director/experiments/tensorflow_mnist/settings.py b/tests/github/interactive_api_director/experiments/tensorflow_mnist/settings.py index 1ad7fa5151d..8903160e150 100644 --- a/tests/github/interactive_api_director/experiments/tensorflow_mnist/settings.py +++ b/tests/github/interactive_api_director/experiments/tensorflow_mnist/settings.py @@ -13,7 +13,7 @@ model = keras.Model(inputs=inputs, outputs=outputs) # Instantiate an optimizer. -optimizer = keras.optimizers.SGD(learning_rate=1e-3) +optimizer = keras.optimizers.legacy.SGD(learning_rate=1e-3) # Instantiate a loss function. loss_fn = keras.losses.SparseCategoricalCrossentropy(from_logits=True) model.compile(optimizer, loss_fn)