Test Custom Step Usecase (aws#331)

atqy · Aug 19, 2020 · 37304c3 · 37304c3
1 parent cb45e75
commit 37304c3
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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 = """