From 9a4d99150a035b8dde7b4f9e72f25745af17b609 Mon Sep 17 00:00:00 2001 From: sasha-gitg <44654632+sasha-gitg@users.noreply.github.com> Date: Tue, 13 Apr 2021 12:31:59 -0400 Subject: [PATCH] fix: make logging more informative during training (#310) --- google/cloud/aiplatform/training_jobs.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/google/cloud/aiplatform/training_jobs.py b/google/cloud/aiplatform/training_jobs.py index 8628452725..220a34637e 100644 --- a/google/cloud/aiplatform/training_jobs.py +++ b/google/cloud/aiplatform/training_jobs.py @@ -646,12 +646,12 @@ def _block_until_complete(self): ) ) log_wait = min(log_wait * multiplier, max_wait) - previous_time = current_time + previous_time = current_time time.sleep(wait) self._raise_failure() - _LOGGER.log_action_completed_against_resource("", "run", self) + _LOGGER.log_action_completed_against_resource("run", "completed", self) if self._gca_resource.model_to_upload and not self.has_failed: _LOGGER.info(