feat(components): Fix typos: 'statgey' -> 'strategy'in v1 GCPC custom…

… job components/utils Signed-off-by: Ze Mao <[email protected]> PiperOrigin-RevId: 725795071
kubeflow · Feb 11, 2025 · dcaf5a4 · dcaf5a4
1 parent 65d1d79
commit dcaf5a4
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/...ts/google-cloud/google_cloud_pipeline_components/container/v1/custom_job/remote_runner.py b/...ts/google-cloud/google_cloud_pipeline_components/container/v1/custom_job/remote_runner.py
@@ -37,12 +37,13 @@ def create_custom_job_with_client(job_client, parent, job_spec):
   # max_wait_duration is acceptable only when strategy is FLEX_START in
   # CustomJob API. Clear max_wait_duration if strategy is not FLEX_START.
   if (
-      'scheduling' in job_spec
-      and 'stategy' in job_spec['scheduling']
-      and job_spec['scheduling']['strategy'] != 'FLEX_START'
-      and 'max_wait_duration' in job_spec['scheduling']
+      'job_spec' in job_spec
+      and 'scheduling' in job_spec['job_spec']
+      and 'strategy' in job_spec['job_spec']['scheduling']
+      and job_spec['job_spec']['scheduling']['strategy'] != 'FLEX_START'
+      and 'max_wait_duration' in job_spec['job_spec']['scheduling']
   ):
-    del job_spec['scheduling']['max_wait_duration']
+    del job_spec['job_spec']['scheduling']['max_wait_duration']
   try:
     create_custom_job_fn = job_client.create_custom_job(
         parent=parent, custom_job=job_spec