From 9ba8df0e231b434cf7546150db0ee202acfdc25d Mon Sep 17 00:00:00 2001 From: Doug Branton Date: Thu, 14 Dec 2023 10:50:26 -0800 Subject: [PATCH] black run --- src/tape/ensemble.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/src/tape/ensemble.py b/src/tape/ensemble.py index 4abe9915..0bf83b91 100644 --- 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