diff --git a/declearn/dataset/tensorflow/__init__.py b/declearn/dataset/tensorflow/__init__.py index ed43e1acbdd3753ed943ae5b50d5758ca82de935..fc2f71ebf8450c57b063e17d06b48f2645634409 100644 --- a/declearn/dataset/tensorflow/__init__.py +++ b/declearn/dataset/tensorflow/__init__.py @@ -17,7 +17,7 @@ """Dataset interface to wrap up 'tensorflow.data.Dataset' instances. -* [TensorflowDataset][declarn.dataset.tensorflow.TensorflowDataset]: +* [TensorflowDataset][declearn.dataset.tensorflow.TensorflowDataset]: Dataset subclass to wrap up 'tensorflow.data.Dataset' instances. """ diff --git a/declearn/dataset/torch/__init__.py b/declearn/dataset/torch/__init__.py index 01e90d6ef6599296c7d6c6536e0d275fea773bf1..048cd7b10cc3eb15f7249eff22ebd57fcb3ef6cc 100644 --- a/declearn/dataset/torch/__init__.py +++ b/declearn/dataset/torch/__init__.py @@ -19,7 +19,7 @@ The main class implementing by this submodule is `TorchDataset`: -* [TorchDataset][declarn.dataset.torch.TorchDataset]: +* [TorchDataset][declearn.dataset.torch.TorchDataset]: Dataset subclass serving torch Datasets. Some utils are also exposed here, either used as part of the `TorchDataset` diff --git a/declearn/dataset/torch/_torch.py b/declearn/dataset/torch/_torch.py index c8e1f47d48843bbe82ffad2e3797e87c2c9e0a3a..c330c4c8666870ccf314ab1adf8c9a2ece442bd4 100644 --- a/declearn/dataset/torch/_torch.py +++ b/declearn/dataset/torch/_torch.py @@ -213,7 +213,7 @@ class TorchDataset(Dataset): @staticmethod def check_dataset_specs(specs) -> None: """Utility function checking that user-defined `get_specs()` - method returns valid [DataSpecs][declearn.dataset.Dataspecs] + method returns valid [DataSpecs][declearn.dataset.DataSpecs] fields.""" acceptable = {f.name for f in dataclasses.fields(DataSpecs)} for key in specs.keys():