Skip to content

Commit 5c993d2

Browse files
jaycee-licopybara-github
authored andcommitted
chore: Disable preview custom job logs when executing remote training
PiperOrigin-RevId: 571095097
1 parent 9fedd34 commit 5c993d2

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

vertexai/preview/_workflow/executor/training.py

+2
Original file line numberDiff line numberDiff line change
@@ -713,6 +713,7 @@ def remote_training(invokable: shared._Invokable, rewrapper: Any):
713713

714714
# disable CustomJob logs
715715
logging.getLogger("google.cloud.aiplatform.jobs").disabled = True
716+
logging.getLogger("google.cloud.aiplatform.preview.jobs").disabled = True
716717
try:
717718
job = jobs.CustomJob(
718719
display_name=display_name,
@@ -746,6 +747,7 @@ def remote_training(invokable: shared._Invokable, rewrapper: Any):
746747
finally:
747748
# enable CustomJob logs after remote training job is done
748749
logging.getLogger("google.cloud.aiplatform.jobs").disabled = False
750+
logging.getLogger("google.cloud.aiplatform.preview.jobs").disabled = False
749751

750752
if job.state in jobs._JOB_ERROR_STATES:
751753
return job

0 commit comments

Comments
 (0)