Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

prevent internal call to has_prepared_data showing a warning (#8271)

This commit is contained in:
Adrian Wälchli 2021-07-05 13:44:33 +02:00 committed by GitHub
parent 51e0d748b4
commit d7a0786661
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch_lightning/trainer/connectors/data_connector.py
View File

@ -60,7 +60,7 @@ class DataConnector:
def can_prepare_data(self):
should_call_dm_prepare_data = True
if self.trainer.datamodule is not None and is_overridden('prepare_data', self.trainer.datamodule):
should_call_dm_prepare_data = not self.trainer.datamodule.has_prepared_data
should_call_dm_prepare_data = not self.trainer.datamodule._has_prepared_data
if self.trainer.prepare_data_per_node:
return self.trainer.local_rank == 0 and should_call_dm_prepare_data