diff --git a/tests/zero_code_change/test_tensorflow2_integration.py b/tests/zero_code_change/test_tensorflow2_integration.py index cfe179a4d5..bc0a4d13e1 100644 --- a/tests/zero_code_change/test_tensorflow2_integration.py +++ b/tests/zero_code_change/test_tensorflow2_integration.py @@ -266,7 +266,6 @@ def test_keras_v2_multi_collections(script_mode, eager_mode): @pytest.mark.parametrize("script_mode", [False]) @pytest.mark.parametrize("eager_mode", [True]) -@pytest.mark.skip def test_keras_v2_custom_train_step(script_mode, eager_mode): # Test multiple collections included in hook json json_file_contents = """