diff --git a/tests/unit/aiplatform/test_explain_saved_model_metadata_builder_tf1_test.py b/tests/unit/aiplatform/test_explain_saved_model_metadata_builder_tf1_test.py index 1300df5968..fc178b1ce5 100644 --- a/tests/unit/aiplatform/test_explain_saved_model_metadata_builder_tf1_test.py +++ b/tests/unit/aiplatform/test_explain_saved_model_metadata_builder_tf1_test.py @@ -26,6 +26,7 @@ from test_models import upload_model_mock, get_model_mock # noqa: F401 +@pytest.mark.usefixtures("google_auth_mock") class SavedModelMetadataBuilderTF1Test(tf.test.TestCase): def _set_up(self): self.sess = tf.Session(graph=tf.Graph()) diff --git a/tests/unit/aiplatform/test_explain_saved_model_metadata_builder_tf2_test.py b/tests/unit/aiplatform/test_explain_saved_model_metadata_builder_tf2_test.py index a18eed243c..82493db24e 100644 --- a/tests/unit/aiplatform/test_explain_saved_model_metadata_builder_tf2_test.py +++ b/tests/unit/aiplatform/test_explain_saved_model_metadata_builder_tf2_test.py @@ -27,6 +27,7 @@ from test_models import upload_model_mock, get_model_mock # noqa: F401 +@pytest.mark.usefixtures("google_auth_mock") class SavedModelMetadataBuilderTF2Test(tf.test.TestCase): def _set_up_sequential(self): # Set up for the sequential. diff --git a/tests/unit/aiplatform/test_training_jobs.py b/tests/unit/aiplatform/test_training_jobs.py index a956584663..47a5ae01a2 100644 --- a/tests/unit/aiplatform/test_training_jobs.py +++ b/tests/unit/aiplatform/test_training_jobs.py @@ -330,6 +330,9 @@ def mock_get_backing_custom_job_with_enable_web_access(): yield get_custom_job_mock +@pytest.mark.skipif( + sys.executable is None, reason="requires python path to invoke subprocess" +) @pytest.mark.usefixtures("google_auth_mock") class TestTrainingScriptPythonPackagerHelpers: def setup_method(self): @@ -446,6 +449,9 @@ def test_get_python_executable_returns_python_executable(self): assert "python" in source_utils._get_python_executable().lower() +@pytest.mark.skipif( + sys.executable is None, reason="requires python path to invoke subprocess" +) @pytest.mark.usefixtures("google_auth_mock") class TestTrainingScriptPythonPackager: def setup_method(self):