File tree 1 file changed +2
-0
lines changed
vertexai/preview/_workflow/executor
1 file changed +2
-0
lines changed Original file line number Diff line number Diff line change @@ -713,6 +713,7 @@ def remote_training(invokable: shared._Invokable, rewrapper: Any):
713
713
714
714
# disable CustomJob logs
715
715
logging .getLogger ("google.cloud.aiplatform.jobs" ).disabled = True
716
+ logging .getLogger ("google.cloud.aiplatform.preview.jobs" ).disabled = True
716
717
try :
717
718
job = jobs .CustomJob (
718
719
display_name = display_name ,
@@ -746,6 +747,7 @@ def remote_training(invokable: shared._Invokable, rewrapper: Any):
746
747
finally :
747
748
# enable CustomJob logs after remote training job is done
748
749
logging .getLogger ("google.cloud.aiplatform.jobs" ).disabled = False
750
+ logging .getLogger ("google.cloud.aiplatform.preview.jobs" ).disabled = False
749
751
750
752
if job .state in jobs ._JOB_ERROR_STATES :
751
753
return job
You can’t perform that action at this time.
0 commit comments