lightning/tests/core
Nathan Raw 036bcea499
Call DataModule hooks implicitly in trainer (#2755)
*  call dm hooks in trainer implicitly

*  update tests

* 📝 remove unused stage arg from dm docs

*  update tests

*  update tests

* 🚧 include stage in datamodule.setup

* 📝 docs

* 📝 docs

* added more dm tests

* added more dm tests

* 🐛 call dm.setup everywhere

* 🔥 pickle tests now implied by accelerator tests

* 🎨 set dm as attr of trainer

* 🐛 .

* 🚧 wip

* add can prepare test

* add can prepare test

* verified setup in fit

* fixed setup call

* fixed setup call

* fixed setup call

Co-authored-by: William Falcon <waf2107@columbia.edu>
2020-08-01 20:17:57 -04:00
..
__init__.py Refactor model summary + generalize example input array (#1773) 2020-06-15 17:05:58 -04:00
test_datamodules.py Call DataModule hooks implicitly in trainer (#2755) 2020-08-01 20:17:57 -04:00
test_decorators.py cleaning tests (#2201) 2020-06-15 22:03:40 -04:00
test_memory.py fix dtype conversion of example_input_array in model summary (#2510) 2020-07-05 07:17:22 -04:00