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

qiaodev · copybara-github · commit d0585e8ed58c · 2024-04-03T14:05:21.000-07:00
PiperOrigin-RevId: 621632653
diff --git a/testing/constraints-3.10.txt b/testing/constraints-3.10.txt
@@ -12,3 +12,4 @@ mlflow==1.30.1 # Pinned to speed up installation
 pytest-xdist==3.3.1 # Pinned to unbreak unit tests
 ray==2.4.0 # Pinned until 2.9.3 is verified for Ray tests
 ipython==8.22.2 # Pinned to unbreak TypeAliasType import error
+scikit-learn!=1.4.1.post1 # Pin to unbreak test_sklearn (b/332610038)
diff --git a/tests/system/vertexai/test_tensorflow.py b/tests/system/vertexai/test_tensorflow.py
@@ -62,6 +62,10 @@ class TestRemoteExecutionTensorflow(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(
