yapf adjustments after yapf --in-place --recursive ./src

JosephTheMoUSE · Feb 29, 2024 · 96df263 · 96df263
1 parent 4e95332
commit 96df263
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 9 deletions.
diff --git a/src/mouse/entropy_energy/entropy_energy.py b/src/mouse/entropy_energy/entropy_energy.py
@@ -58,8 +58,8 @@ def _mark_high_energy_rows(spectrogram, energy_window, energy_threshold):
         spectrogram = spectrogram.numpy()
     mean_energy = np.mean(np.power(spectrogram, 2))
     width = spectrogram.shape[1]
-    high_energy = (_moving_average_2d(np.power(spectrogram, 2), energy_window, width) >
-                   energy_threshold * mean_energy)
+    high_energy = (_moving_average_2d(np.power(spectrogram, 2), energy_window, width)
+                   > energy_threshold * mean_energy)
     return _binary_mask_to_ranges(high_energy)
 
 

diff --git a/src/mouse/segmentation/GAC.py b/src/mouse/segmentation/GAC.py
@@ -132,9 +132,7 @@ def _iter_callback(level_set):
             _kwargs['iter_callback'] = _iter_callback
 
         level_set_result = segmentation.morphological_geodesic_active_contour(
-            _spec,
-            init_level_set=level_set_init,
-            **_kwargs)
+            _spec, init_level_set=level_set_init, **_kwargs)
 
     boxes = data_util.find_bounding_boxes(level_set_result,
                                           min_side_length=min_side_length)

diff --git a/src/mouse/utils/metrics.py b/src/mouse/utils/metrics.py
@@ -88,8 +88,10 @@ def _to_bit_map(
     bit_map = np.zeros((height, width), dtype=bool)
 
     for squeak in squeaks:
-        bit_map[squeak.freq_start - freq_start:squeak.freq_end + 1 - freq_start,
-                squeak.t_start - t_start:squeak.t_end + 1 - t_start,] = True
+        bit_map[
+            squeak.freq_start - freq_start:squeak.freq_end + 1 - freq_start,
+            squeak.t_start - t_start:squeak.t_end + 1 - t_start,
+        ] = True
 
     return bit_map
 
@@ -201,8 +203,9 @@ def intersection_over_union_elementwise(
     target_sorted = sorted(target, key=lambda squeak: (squeak.t_start, squeak.t_end))
     cover_sorted = sorted(cover, key=lambda squeak: (squeak.t_end, squeak.t_start))
 
-    result: Dict[data_util.SqueakBox,
-                 dict] = {target_squeak: dict() for target_squeak in target_sorted}
+    result: Dict[data_util.SqueakBox, dict] = {
+        target_squeak: dict() for target_squeak in target_sorted
+    }
 
     j_start = 0
     for target_squeak in target_sorted: