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