[CodeStyle] Address linter findings.

Author: gmarciani

tests/integration-tests/tests/common/utils.py

@@ -563,13 +563,19 @@ def get_capacity_reservation_id(request, instance_type, region, count, os):
     reservations_ids = []
     ec2_client = boto3.client("ec2", region_name=region)
     if request.config.getoption("capacity_reservation_id"):
-        capacity_reservation = ec2_client.describe_capacity_reservations(CapacityReservationIds=[request.config.getoption("capacity_reservation_id")])
+        capacity_reservation = ec2_client.describe_capacity_reservations(
+            CapacityReservationIds=[request.config.getoption("capacity_reservation_id")]
+        )
         if capacity_reservation:
             reservations_ids.append(
                 {
-                    "CapacityReservationId": capacity_reservation.get("CapacityReservations", [])[0]["CapacityReservationId"],
+                    "CapacityReservationId": capacity_reservation.get("CapacityReservations", [])[0][
+                        "CapacityReservationId"
+                    ],
                     "TotalInstanceCount": capacity_reservation.get("CapacityReservations", [])[0]["TotalInstanceCount"],
-                    "AvailableInstanceCount": capacity_reservation.get("CapacityReservations", [])[0]["AvailableInstanceCount"],
+                    "AvailableInstanceCount": capacity_reservation.get("CapacityReservations", [])[0][
+                        "AvailableInstanceCount"
+                    ],
                 }
             )
     else:

tests/integration-tests/tests/efa/test_efa.py

@@ -58,7 +58,7 @@ def test_efa(
     max_queue_size = 2
     p6_b200_capacity_reservation_id = None
     if instance == "p6-b200.48xlarge":
-        capacity_reservations_ids = get_capacity_reservation_id(request,instance, region, max_queue_size, os)
+        capacity_reservations_ids = get_capacity_reservation_id(request, instance, region, max_queue_size, os)
         if capacity_reservations_ids:
             p6_b200_capacity_reservation_id = capacity_reservations_ids[0].get("CapacityReservationId")
         else:

tests/integration-tests/tests/ultraserver/test_gb200.py

@@ -41,7 +41,11 @@
 def submit_job_imex_status(rce: RemoteCommandExecutor, queue: str, max_nodes: int = 1, ignore_imex_check: bool = False):
     logging.info("Submitting job to check IMEX status")
     slurm = SlurmCommands(rce)
-    job_command = "/opt/parallelcluster/shared/nvidia-imex-status.job ignore-imex-check" if ignore_imex_check else "/opt/parallelcluster/shared/nvidia-imex-status.job"
+    job_command = (
+        "/opt/parallelcluster/shared/nvidia-imex-status.job ignore-imex-check"
+        if ignore_imex_check
+        else "/opt/parallelcluster/shared/nvidia-imex-status.job"
+    )
     job_id = slurm.submit_command_and_assert_job_accepted(
         submit_command_args={
             "command": job_command,