Skip to content

Commit

Permalink
Docs
Browse files Browse the repository at this point in the history
  • Loading branch information
carmocca committed Jul 5, 2021
1 parent 22120c2 commit 419838e
Showing 1 changed file with 2 additions and 5 deletions.
7 changes: 2 additions & 5 deletions pytorch_lightning/core/datamodule.py
Original file line number Diff line number Diff line change
Expand Up @@ -62,10 +62,7 @@ def teardown(self):
* **test_dataloader** the test dataloader(s).
* **teardown** (things to do on every accelerator in distributed mode when finished)
This allows you to share a full dataset without explaining how to download,
split transform and process the data
This allows you to share a full dataset without explaining how to download, split, transform, and process the data
"""

name: str = ...
Expand Down Expand Up @@ -380,7 +377,7 @@ def test_dataloader():

def __new__(cls, *args: Any, **kwargs: Any) -> 'LightningDataModule':
obj = super().__new__(cls)
# track `DataHooks` calls and run `prepare_data` only on rank zero
# track `DataHooks` calls
obj.prepare_data = cls._track_data_hook_calls(obj, obj.prepare_data)
obj.setup = cls._track_data_hook_calls(obj, obj.setup)
obj.teardown = cls._track_data_hook_calls(obj, obj.teardown)
Expand Down

0 comments on commit 419838e

Please sign in to comment.