Fix Ruff rule B017

Author: smokestacklightnin

tfx/extensions/google_cloud_ai_platform/bulk_inferrer/executor_test.py

@@ -235,7 +235,7 @@ def testDoFailedModelDeployment(self, mock_runner, mock_run_model_inference,
     mock_runner.create_model_for_aip_prediction_if_not_exist.return_value = True
 
     bulk_inferrer = executor.Executor(self._context)
-    with self.assertRaises(Exception):
+    with self.assertRaises(Exception): # noqa: B017
       bulk_inferrer.Do(input_dict, output_dict, exec_properties)
 
     mock_runner.delete_model_from_aip_if_exists.assert_called_once_with(

tfx/orchestration/experimental/core/pipeline_ops_test.py

@@ -3803,7 +3803,7 @@ def test_orchestrate_pipelines_with_not_recoverable_error_from_MLMD(
 
     with self._mlmd_cm as mlmd_connection_manager:
       task_queue = tq.TaskQueue()
-      with self.assertRaises(Exception):
+      with self.assertRaises(Exception): # noqa: B017
         pipeline_ops.orchestrate(
             mlmd_connection_manager,
             task_queue,