diff --git a/src/tape/ensemble.py b/src/tape/ensemble.py index 3aa5e442..6561fa5a 100644 --- a/src/tape/ensemble.py +++ b/src/tape/ensemble.py @@ -473,11 +473,13 @@ def persist(self, **kwargs): self.update_frame(self.source.persist(**kwargs)) def sample_objects(self, **kwargs): - """Selects a sample of objects. + """Selects a random sample of objects. This sampling will be lazily applied to the SourceFrame as well. A new Ensemble object is created, and no additional EnsembleFrames will be - carried into the new Ensemble object. + carried into the new Ensemble object. See + https://docs.dask.org/en/latest/generated/dask.dataframe.DataFrame.sample.html + for details on available kwargs. Parameters ---------- @@ -512,6 +514,7 @@ def sample_objects(self, **kwargs): self.cleanup_client = False new_ens.cleanup_client = False else: + print("here") new_ens = Ensemble(client=False) new_ens.update_frame(object_subset)