Skip to content

Commit

Permalink
Standardize how clients are instantiated
Browse files Browse the repository at this point in the history
The `_detect_mount_points_from_batch_job()` function is obtaining its
client in a slightly different way compared to the lambda version
(`_detect_mount_points_from_lambda()`).

This commit just standardizes the client instantiation method so that
the batch version matches the lambda version of the function.
  • Loading branch information
njmei committed Dec 17, 2024
1 parent e9d20c6 commit c1619a4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/aibs_informatics_aws_utils/efs/mount_point.py
Original file line number Diff line number Diff line change
Expand Up @@ -460,7 +460,7 @@ def _detect_mount_points_from_lambda(lambda_function_name: str) -> List[MountPoi

def _detect_mount_points_from_batch_job(batch_job_id: str) -> List[MountPointConfiguration]:
mount_points: List[MountPointConfiguration] = []
batch = AWSService.BATCH.get_client()
batch = get_batch_client()
response = batch.describe_jobs(jobs=[batch_job_id])
job_container = response.get("jobs", [{}])[0].get("container", {})
batch_mount_points = job_container.get("mountPoints")
Expand Down

0 comments on commit c1619a4

Please sign in to comment.