diff --git a/src/tape/ensemble.py b/src/tape/ensemble.py index 06fda41d..d1208767 100644 --- a/src/tape/ensemble.py +++ b/src/tape/ensemble.py @@ -1226,12 +1226,7 @@ def _batch_apply(df, func, on, *args, **kwargs): ) # Output standardization - batch = self._standardize_batch(batch, on, by_band) - # Inherit divisions if known from source and the resulting index is the id - # Groupby on index should always return a subset that adheres to the same divisions criteria - if self.source.known_divisions and batch.index.name == self._id_col: - batch.divisions = self.source.divisions if label is not None: if label == "": @@ -2473,11 +2468,6 @@ def sf2(self, sf_method="basic", argument_container=None, use_map=True): else: result = self.batch(calc_sf2, use_map=use_map, argument_container=argument_container) - # Inherit divisions information if known - if self.source.known_divisions and self.object.known_divisions: - pass # TODO: Can no longer directly set divisions - # result.divisions = self.source.divisions - return result def _translate_meta(self, meta):