From 3a3cd9ef714564eda72d3bcd2897840071bf9d72 Mon Sep 17 00:00:00 2001 From: Sasha Sobran Date: Tue, 13 Apr 2021 11:33:29 -0400 Subject: [PATCH 1/2] fix: save previous logged time only when logging --- google/cloud/aiplatform/training_jobs.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/google/cloud/aiplatform/training_jobs.py b/google/cloud/aiplatform/training_jobs.py index 8628452725..a34ab3f285 100644 --- a/google/cloud/aiplatform/training_jobs.py +++ b/google/cloud/aiplatform/training_jobs.py @@ -646,7 +646,7 @@ 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() From 16692e66e670639198fdcee65c48bab7162d06fa Mon Sep 17 00:00:00 2001 From: Sasha Sobran Date: Tue, 13 Apr 2021 11:37:30 -0400 Subject: [PATCH 2/2] fix: make run completion more informative --- google/cloud/aiplatform/training_jobs.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/google/cloud/aiplatform/training_jobs.py b/google/cloud/aiplatform/training_jobs.py index a34ab3f285..220a34637e 100644 --- a/google/cloud/aiplatform/training_jobs.py +++ b/google/cloud/aiplatform/training_jobs.py @@ -651,7 +651,7 @@ def _block_until_complete(self): 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(