Skip to content

Commit

Permalink
update docstring
Browse files Browse the repository at this point in the history
  • Loading branch information
dougbrn committed Jan 31, 2024
1 parent 3f35e03 commit 9945c81
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions src/tape/ensemble.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
----------
Expand Down Expand Up @@ -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)
Expand Down

0 comments on commit 9945c81

Please sign in to comment.