diff --git a/proteinshake/database.py b/proteinshake/database.py index aae828ce..ec3c7fef 100644 --- a/proteinshake/database.py +++ b/proteinshake/database.py @@ -91,9 +91,8 @@ def __init__(self) -> None: class Collection: """ - A set of Proteins as a result of a Database query. Provides methods to store and load to the filesystem. + A set of Proteins as a result of a Database query. Provides methods to store to the filesystem. Stores files as tar bundles with sharded structures, a protein-level info file, metadata, and assets. - Loads files as an iterator with associated metadata and assets. """ def __init__(self) -> None: @@ -119,7 +118,8 @@ def __init__(self) -> None: class Dataset: """ Abstract class to define a Database query and a file path. Used to create and load a dataset. - Provides the means to download a precomputed Collection from a server or filesystem. + Provides the means to (down-)load a precomputed Collection from a server or filesystem. + Provides a ProteinIterator for the Task. """ def __init__(self) -> None: