feat: reraise exceptions from API calls

dizcology · copybara-github · commit d72bc835184c · 2022-12-12T11:00:06.000-08:00
PiperOrigin-RevId: 494773031
diff --git a/google/cloud/aiplatform/datasets/column_names_dataset.py b/google/cloud/aiplatform/datasets/column_names_dataset.py
@@ -157,7 +157,7 @@ def _retrieve_gcs_source_columns(
                 "There was a problem extracting the headers from the CSV file at '{}': {}".format(
                     gcs_csv_file_path, err
                 )
-            )
+            ) from err
         finally:
             logger.removeFilter(logging_warning_filter)
 
diff --git a/google/cloud/aiplatform/initializer.py b/google/cloud/aiplatform/initializer.py
@@ -212,8 +212,8 @@ def project(self) -> str:
 
         try:
             _, project_id = google.auth.default()
-        except GoogleAuthError:
-            raise GoogleAuthError(project_not_found_exception_str)
+        except GoogleAuthError as exc:
+            raise GoogleAuthError(project_not_found_exception_str) from exc
 
         if not project_id:
             raise ValueError(project_not_found_exception_str)
diff --git a/google/cloud/aiplatform/models.py b/google/cloud/aiplatform/models.py
@@ -2508,10 +2508,10 @@ def training_job(self) -> Optional["aiplatform.training_jobs._TrainingJob"]:
                 location=self.location,
                 credentials=self.credentials,
             )
-        except api_exceptions.NotFound:
+        except api_exceptions.NotFound as exc:
             raise api_exceptions.NotFound(
                 f"The training job used to create this model could not be found: {job_name}"
-            )
+            ) from exc
 
     @property
     def container_spec(self) -> Optional[model_v1.ModelContainerSpec]:
diff --git a/google/cloud/aiplatform/tensorboard/uploader.py b/google/cloud/aiplatform/tensorboard/uploader.py
@@ -873,7 +873,7 @@ def flush(self):
                         hasattr(e, "code")
                         and getattr(e, "code")() == grpc.StatusCode.NOT_FOUND
                     ):
-                        raise ExperimentNotFoundError()
+                        raise ExperimentNotFoundError() from e
                     logger.error("Upload call failed with error %s", e)
 
         self._new_request()
@@ -1161,7 +1161,7 @@ def _validate(
                 "a bug in the process that wrote the tensor.\n\n"
                 "The tensor has tag '%s' and is at step %d and wall_time %.6f.\n\n"
                 "Original error:\n%s" % (value.tag, event.step, event.wall_time, error)
-            )
+            ) from error
         return True
 
 
diff --git a/google/cloud/aiplatform/tensorboard/uploader_main.py b/google/cloud/aiplatform/tensorboard/uploader_main.py
@@ -110,7 +110,7 @@ def main(argv):
             raise app.UsageError(
                 "Tensorboard resource %s not found" % FLAGS.tensorboard_resource_name,
                 exitcode=0,
-            )
+            ) from rpc_error
         raise
 
     if tensorboard.blob_storage_path_prefix:

Original file line number	Diff line number	Diff line change
`@@ -157,7 +157,7 @@ def _retrieve_gcs_source_columns(`
`157`	`157`	`"There was a problem extracting the headers from the CSV file at '{}': {}".format(`
`158`	`158`	`gcs_csv_file_path, err`
`159`	`159`	`)`
`160`		`- )`
	`160`	`+ ) from err`
`161`	`161`	`finally:`
`162`	`162`	`logger.removeFilter(logging_warning_filter)`
`163`	`163`