From 37304c3f0456c468963fe15d166e0cbf9917027c Mon Sep 17 00:00:00 2001 From: Nihal Harish Date: Tue, 18 Aug 2020 18:44:31 -0700 Subject: [PATCH] Test Custom Step Usecase (#331) --- tests/zero_code_change/test_tensorflow2_integration.py | 1 - 1 file changed, 1 deletion(-) 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 = """