fix: make logging more informative during training (#310)

sasha-gitg · web-flow · commit 9a4d99150a03 · 2021-04-13T12:31:59.000-04:00
diff --git 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(

Original file line number	Diff line number	Diff line change
`@@ -646,12 +646,12 @@ def _block_until_complete(self):`
`646`	`646`	`)`
`647`	`647`	`)`
`648`	`648`	`log_wait = min(log_wait * multiplier, max_wait)`
`649`		`- previous_time = current_time`
	`649`	`+ previous_time = current_time`
`650`	`650`	`time.sleep(wait)`
`651`	`651`
`652`	`652`	`self._raise_failure()`
`653`	`653`
`654`		`- _LOGGER.log_action_completed_against_resource("", "run", self)`
	`654`	`+ _LOGGER.log_action_completed_against_resource("run", "completed", self)`
`655`	`655`
`656`	`656`	`if self._gca_resource.model_to_upload and not self.has_failed:`
`657`	`657`	`_LOGGER.info(`