Skip to content

Commit 6a00066

Browse files
awaelchlilexierule
authored andcommitted
Fix data fetcher selection (#11294)
1 parent f562427 commit 6a00066

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

CHANGELOG.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
1313

1414
- Fixed `LightningCLI` race condition while saving the config ([#11199](https://github.com/PyTorchLightning/pytorch-lightning/pull/11199))
1515

16+
- Fixed data fetcher selection ([#11294](https://github.com/PyTorchLightning/pytorch-lightning/pull/11294))
1617
## [1.5.7] - 2021-12-21
1718

1819
### Fixed

pytorch_lightning/trainer/connectors/data_connector.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,7 @@ def _select_data_fetcher(self) -> AbstractDataFetcher:
120120

121121
def get_profiled_dataloader(self, dataloader: Iterable, dataloader_idx: int = 0) -> Iterable:
122122
stage: str = self.trainer.state.stage.value
123-
data_fetcher = setattr(self, f"{stage}_data_fetcher", None) or self._select_data_fetcher()
123+
data_fetcher = getattr(self, f"{stage}_data_fetcher", None) or self._select_data_fetcher()
124124
data_fetcher.setup(
125125
dataloader,
126126
stage=stage,

0 commit comments

Comments
 (0)