Updated e2e tabular test

ivanmkc · ivanmkc · commit 804eb0662c5c · 2022-01-31T14:30:10.000-05:00
diff --git 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])

Original file line number	Diff line number	Diff line change
`@@ -159,7 +159,7 @@ def test_end_to_end_tabular(self, shared_state):`
`159`	`159`	`custom_job._gca_resource.training_task_inputs["scheduling"][`
`160`	`160`	`"restartJobOnWorkerRestart"`
`161`	`161`	`]`
`162`		`- == True`
	`162`	`+ is True`
`163`	`163`	`)`
`164`	`164`
`165`	`165`	`custom_prediction = custom_endpoint.predict([_INSTANCE])`