[CLEAN] ruff reset

prs-eth · Apr 15, 2024 · 0cb80c8 · 0cb80c8
1 parent fb65871
commit 0cb80c8
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/dataset/base_depth_dataset.py b/src/dataset/base_depth_dataset.py
@@ -200,13 +200,13 @@ def _training_preprocess(self, rasters):
         # Set invalid pixel to far plane
         if self.move_invalid_to_far_plane:
             if self.depth_transform.far_plane_at_max:
-                rasters["depth_filled_norm"][
-                    ~rasters["valid_mask_filled"]
-                ] = self.depth_transform.norm_max
+                rasters["depth_filled_norm"][~rasters["valid_mask_filled"]] = (
+                    self.depth_transform.norm_max
+                )
             else:
-                rasters["depth_filled_norm"][
-                    ~rasters["valid_mask_filled"]
-                ] = self.depth_transform.norm_min
+                rasters["depth_filled_norm"][~rasters["valid_mask_filled"]] = (
+                    self.depth_transform.norm_min
+                )
 
         # Resize
         if self.resize_to_hw is not None: