wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-12 12:40:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update type hints for multiple dataloaders in .fit() and .test() (#1723)

Browse Source 
* update typehints

* change log
This commit is contained in:
Adrian Wälchli
2020-05-04 08:24:34 -04:00
committed by GitHub
parent 0cd5e64701
commit d28b145393
3 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/trainer/test_dataloaders.py
+3 -3
View File
@@ -113,7 +113,7 @@ def test_multiple_test_dataloader(tmpdir):
trainer.fit(model)
trainer.test()
# verify there are 2 val loaders
# verify there are 2 test loaders
assert len(trainer.test_dataloaders) == 2, \
'Multiple test_dataloaders not initiated properly'
@@ -125,7 +125,7 @@ def test_multiple_test_dataloader(tmpdir):
trainer.test()
def test_train_dataloaders_passed_to_fit(tmpdir):
def test_train_dataloader_passed_to_fit(tmpdir):
"""Verify that train dataloader can be passed to fit """
class CurrentTestModel(LightTrainDataloader, TestModelBase):
@@ -175,7 +175,7 @@ def test_train_val_dataloaders_passed_to_fit(tmpdir):
def test_all_dataloaders_passed_to_fit(tmpdir):
"""Verify train, val & test dataloader can be passed to fit """
"""Verify train, val & test dataloader(s) can be passed to fit and test method"""
class CurrentTestModel(
LightTrainDataloader,