chore: Disable preview custom job logs when executing remote training

PiperOrigin-RevId: 571095097
googleapis · Oct 5, 2023 · 5c993d2 · 5c993d2
1 parent 9fedd34
commit 5c993d2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/vertexai/preview/_workflow/executor/training.py b/vertexai/preview/_workflow/executor/training.py
@@ -713,6 +713,7 @@ def remote_training(invokable: shared._Invokable, rewrapper: Any):
 
     # disable CustomJob logs
     logging.getLogger("google.cloud.aiplatform.jobs").disabled = True
+    logging.getLogger("google.cloud.aiplatform.preview.jobs").disabled = True
     try:
         job = jobs.CustomJob(
             display_name=display_name,
@@ -746,6 +747,7 @@ def remote_training(invokable: shared._Invokable, rewrapper: Any):
     finally:
         # enable CustomJob logs after remote training job is done
         logging.getLogger("google.cloud.aiplatform.jobs").disabled = False
+        logging.getLogger("google.cloud.aiplatform.preview.jobs").disabled = False
 
     if job.state in jobs._JOB_ERROR_STATES:
         return job