clean up

googleapis · Apr 5, 2022 · 3914dd0 · 3914dd0
1 parent a34e3a4
commit 3914dd0
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/tests/unit/aiplatform/test_pipeline_jobs.py b/tests/unit/aiplatform/test_pipeline_jobs.py
@@ -354,7 +354,7 @@ def test_run_call_pipeline_service_create(
         runtime_config = gca_pipeline_job_v1.PipelineJob.RuntimeConfig()._pb
         json_format.ParseDict(expected_runtime_config_dict, runtime_config)
 
-        job_spec = job_spec if isinstance(job_spec, dict) else yaml.safe_load(job_spec)
+        job_spec = yaml.safe_load(job_spec)
         pipeline_spec = job_spec.get("pipelineSpec") or job_spec
 
         # Construct expected request
@@ -435,7 +435,7 @@ def test_run_call_pipeline_service_create_with_timeout(
         runtime_config = gca_pipeline_job_v1.PipelineJob.RuntimeConfig()._pb
         json_format.ParseDict(expected_runtime_config_dict, runtime_config)
 
-        job_spec = job_spec if isinstance(job_spec, dict) else yaml.safe_load(job_spec)
+        job_spec = yaml.safe_load(job_spec)
         pipeline_spec = job_spec.get("pipelineSpec") or job_spec
 
         # Construct expected request
@@ -516,7 +516,7 @@ def test_run_call_pipeline_service_create_legacy(
         runtime_config = gca_pipeline_job_v1.PipelineJob.RuntimeConfig()._pb
         json_format.ParseDict(expected_runtime_config_dict, runtime_config)
 
-        job_spec = job_spec if isinstance(job_spec, dict) else yaml.safe_load(job_spec)
+        job_spec = yaml.safe_load(job_spec)
         pipeline_spec = job_spec.get("pipelineSpec") or job_spec
 
         # Construct expected request
@@ -597,7 +597,7 @@ def test_run_call_pipeline_service_create_tfx(
         runtime_config = gca_pipeline_job_v1.PipelineJob.RuntimeConfig()._pb
         json_format.ParseDict(expected_runtime_config_dict, runtime_config)
 
-        job_spec = job_spec if isinstance(job_spec, dict) else yaml.safe_load(job_spec)
+        job_spec = yaml.safe_load(job_spec)
         pipeline_spec = job_spec.get("pipelineSpec") or job_spec
 
         # Construct expected request
@@ -669,7 +669,7 @@ def test_submit_call_pipeline_service_pipeline_job_create(
         runtime_config = gca_pipeline_job_v1.PipelineJob.RuntimeConfig()._pb
         json_format.ParseDict(expected_runtime_config_dict, runtime_config)
 
-        job_spec = job_spec if isinstance(job_spec, dict) else yaml.safe_load(job_spec)
+        job_spec = yaml.safe_load(job_spec)
         pipeline_spec = job_spec.get("pipelineSpec") or job_spec
 
         # Construct expected request
@@ -782,7 +782,7 @@ def test_submit_call_pipeline_service_pipeline_job_create_legacy(
         runtime_config = gca_pipeline_job_v1.PipelineJob.RuntimeConfig()._pb
         json_format.ParseDict(expected_runtime_config_dict, runtime_config)
 
-        job_spec = job_spec if isinstance(job_spec, dict) else yaml.safe_load(job_spec)
+        job_spec = yaml.safe_load(job_spec)
         pipeline_spec = job_spec.get("pipelineSpec") or job_spec
 
         # Construct expected request