black run

lincc-frameworks · Dec 15, 2023 · 0efba3d · 0efba3d
1 parent 236390d
commit 0efba3d
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/src/tape/ensemble.py b/src/tape/ensemble.py
@@ -1174,9 +1174,11 @@ def s2n_inter_quartile_range(flux, err):
                 # interpretted by dask as a single "index" column
                 batch._meta = TapeFrame(columns=on + ["result"])
                 if by_band:
-                    batch = EnsembleFrame.from_dask_dataframe(batch.categorize("band").pivot_table(
-                        index=on[0], columns=self._band_col, aggfunc="sum"
-                    ))
+                    batch = EnsembleFrame.from_dask_dataframe(
+                        batch.categorize("band").pivot_table(
+                            index=on[0], columns=self._band_col, aggfunc="sum"
+                        )
+                    )
 
                     # Need to once again reestablish meta for the pivot
                     band_labels = batch.columns.values
@@ -1203,7 +1205,9 @@ def s2n_inter_quartile_range(flux, err):
                 if by_band:
                     batch = batch.categorize("band")
                     print(isinstance(batch, EnsembleFrame))
-                    batch = EnsembleFrame.from_dask_dataframe(batch.pivot_table(index=on[0], columns=self._band_col, aggfunc="sum"))
+                    batch = EnsembleFrame.from_dask_dataframe(
+                        batch.pivot_table(index=on[0], columns=self._band_col, aggfunc="sum")
+                    )
                     print(isinstance(batch, EnsembleFrame))
 
                     # Need to once again reestablish meta for the pivot