diff --git a/tests/tensorflow2/test_keras_mirrored.py b/tests/tensorflow2/test_keras_mirrored.py index b4394f525..e37d9a45f 100644 --- a/tests/tensorflow2/test_keras_mirrored.py +++ b/tests/tensorflow2/test_keras_mirrored.py @@ -366,7 +366,7 @@ def test_include_regex(out_dir, tf_eager_mode, workers): ) -@pytest.mark.slow +@pytest.mark.skip # Skipping this test for 0.8.1 because it is flaky @pytest.mark.parametrize("workers", ["one", "all"]) def test_include_regex_opt_var(out_dir, tf_eager_mode, workers): include_collections = ["custom_optimizer_variables"]