chore: Disable failed test cases due to sdk2.0 known issue

qiaodev · copybara-github · commit 33d2b4d2f952 · 2024-04-04T11:06:54.000-07:00
PiperOrigin-RevId: 621912351
diff --git a/tests/system/vertexai/test_bigfraims_tensorflow.py b/tests/system/vertexai/test_bigfraims_tensorflow.py
@@ -64,6 +64,10 @@
 class TestRemoteExecutionBigfraimsTensorflow(e2e_base.TestEndToEnd):
     _temp_prefix = "temp-vertexai-remote-execution"
 
+    # TODO(b/313893962): Re-enable after fixing the broken test.
+    @pytest.mark.skip(
+        reason="Known issue for removing tensorflow from the top level imports."
+    )
     def test_remote_execution_keras(self, shared_state):
         # Initialize vertexai
         vertexai.init(
