Minor codacy fix

NKI-AI · May 20, 2024 · c75a838 · c75a838
1 parent bbcdc5a
commit c75a838
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/direct/nn/vsharp/vsharp_engine.py b/direct/nn/vsharp/vsharp_engine.py
@@ -436,10 +436,10 @@ def _do_iteration(
                     # Initialize auxiliary loss weights with a single value of 1.0 if single step
                     auxiliary_loss_weights = torch.ones(1).to(output_images[0])
 
-                for i, _ in enumerate(output_images):
+                for i, output_image in enumerate(output_images):
                     # Data consistency
                     output_kspace = T.apply_padding(
-                        kspace + self._forward_operator(output_images[i], data["sensitivity_map"], ~mask),
+                        kspace + self._forward_operator(output_image, data["sensitivity_map"], ~mask),
                         padding=data.get("padding", None),
                     )
                     # Project predicted k-space onto target k-space if SSL
@@ -655,10 +655,10 @@ def _do_iteration(
                     # Initialize auxiliary loss weights with a single value of 1.0 if single step
                     auxiliary_loss_weights = torch.ones(1).to(output_images[0])
 
-                for i, _ in enumerate(output_images):
+                for i, output_image in enumerate(output_images):
                     # Data consistency
                     output_kspace = T.apply_padding(
-                        kspace + self._forward_operator(output_images[i], data["sensitivity_map"], ~mask),
+                        kspace + self._forward_operator(output_image, data["sensitivity_map"], ~mask),
                         padding=data.get("padding", None),
                     )
                     if is_ssl: