From 3c3b9621079a79de3e29dd31502061dc2c9b79e0 Mon Sep 17 00:00:00 2001 From: Rasmus Oersoe Date: Fri, 13 Sep 2024 14:32:08 +0200 Subject: [PATCH] adjust doc string --- src/graphnet/data/curated_datamodule.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/graphnet/data/curated_datamodule.py b/src/graphnet/data/curated_datamodule.py index fe762a6b9..60fa7e127 100644 --- a/src/graphnet/data/curated_datamodule.py +++ b/src/graphnet/data/curated_datamodule.py @@ -309,7 +309,7 @@ def __init__( validation_dataloader_kwargs: Dict[str, Any] = None, test_dataloader_kwargs: Dict[str, Any] = None, ) -> None: - """Download a public dataset and build DataLoaders get dataloaders. + """Download a public dataset and build DataLoaders. The Dataset can be instatiated in three modes: "train", "test" or "test-no-noise". When instantiated in "train" mode, input data is