fix(job_attachments): pass original exception to AssetSyncError

Signed-off-by: Gahyun Suh <[email protected]>
aws-deadline · Apr 3, 2024 · d60f1ba · d60f1ba
1 parent c3383c2
commit d60f1ba
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/src/deadline/job_attachments/download.py b/src/deadline/job_attachments/download.py
@@ -122,7 +122,7 @@ def get_manifest_from_s3(
             error_details=str(bce),
         ) from bce
     except Exception as e:
-        raise AssetSyncError from e
+        raise AssetSyncError(e) from e
 
 
 def _get_output_manifest_prefix(
@@ -220,7 +220,7 @@ def _get_tasks_manifests_keys_from_s3(
     except JobAttachmentsError:
         raise  # pass along JobAttachmentsErrors if we get them
     except Exception as e:
-        raise AssetSyncError from e
+        raise AssetSyncError(e) from e
 
     # 2. Select all files in the last subfolder (alphabetically) under each "task-{any}" folder.
     for task_folder, files in task_prefixes.items():
@@ -516,7 +516,7 @@ def process_client_error(exc: ClientError, status_code: int):
             error_details=str(bce),
         ) from bce
     except Exception as e:
-        raise AssetSyncError from e
+        raise AssetSyncError(e) from e
 
     download_logger.debug(f"Downloaded {file.path} to {str(local_file_name)}")
     os.utime(local_file_name, (modified_time_override, modified_time_override))  # type: ignore[arg-type]
@@ -646,7 +646,7 @@ def _get_asset_root_from_s3(
             error_details=str(bce),
         ) from bce
     except Exception as e:
-        raise AssetSyncError from e
+        raise AssetSyncError(e) from e
 
     return head["Metadata"].get("asset-root", None)
 

diff --git a/src/deadline/job_attachments/upload.py b/src/deadline/job_attachments/upload.py
@@ -432,7 +432,7 @@ def handler(bytes_uploaded):
                     error_details=str(bce),
                 ) from bce
             except Exception as e:
-                raise AssetSyncError from e
+                raise AssetSyncError(e) from e
 
     @contextmanager
     def _open_non_symlink_file_binary(
@@ -590,7 +590,7 @@ def file_already_uploaded(self, bucket: str, key: str) -> bool:
                 error_details=str(bce),
             ) from bce
         except Exception as e:
-            raise AssetSyncError from e
+            raise AssetSyncError(e) from e
 
     def upload_bytes_to_s3(
         self,
@@ -644,7 +644,7 @@ def upload_bytes_to_s3(
                 error_details=str(bce),
             ) from bce
         except Exception as e:
-            raise AssetSyncError from e
+            raise AssetSyncError(e) from e
 
 
 class S3AssetManager: