From 804eb0662c5c9365812eb38a5b07e5fa487acc62 Mon Sep 17 00:00:00 2001 From: ivanmkc Date: Mon, 31 Jan 2022 14:30:10 -0500 Subject: [PATCH] Updated e2e tabular test --- tests/system/aiplatform/test_e2e_tabular.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/system/aiplatform/test_e2e_tabular.py b/tests/system/aiplatform/test_e2e_tabular.py index 0c6967f451..06b6c4d7c2 100644 --- a/tests/system/aiplatform/test_e2e_tabular.py +++ b/tests/system/aiplatform/test_e2e_tabular.py @@ -159,7 +159,7 @@ def test_end_to_end_tabular(self, shared_state): custom_job._gca_resource.training_task_inputs["scheduling"][ "restartJobOnWorkerRestart" ] - == True + is True ) custom_prediction = custom_endpoint.predict([_INSTANCE])