lightning/tests/utilities
Luis Perez 000fbe63d3
Expose `extract_batch_size` method and add corresponding tests. (#8357)
* expose extract_batch and make public

* first pass

* early return

* add changelog

* move to utilities/data.py

* add test_data.py

* tests are passing

* precommit hook

* address pep8 failure

Co-authored-by: Carlos Mocholi <carlossmocholi@gmail.com>
2021-07-13 11:35:10 +00:00
..
__init__.py New metric classes (#1326) (#1877) 2020-05-19 11:05:07 -04:00
distributed.py Support calling fit and test scripts using "python -m" module syntax with DDP (#8073) 2021-06-23 02:42:04 +00:00
test_all_gather_grad.py Automatically find and run special tests (#6669) 2021-03-26 17:04:59 +00:00
test_apply_func.py Iterate dictionary directly (#8155) 2021-06-27 21:55:16 +02:00
test_apply_func_torchtext.py fix importing torchtext batch (#6365) 2021-03-05 21:39:52 +01:00
test_argparse.py Fix argparse docs (#7148) 2021-04-22 02:13:00 +02:00
test_auto_restart.py [Feat] Add FastForwardSampler 2/n - Fault Tolerant Training (#8307) 2021-07-07 20:21:21 +00:00
test_cli.py LightningCLI documentation improvements (#8303) 2021-07-08 12:35:26 +05:30
test_data.py Expose `extract_batch_size` method and add corresponding tests. (#8357) 2021-07-13 11:35:10 +00:00
test_distributed.py [fix] Better support for rank_zero_only setting for SLURM and torchelastic (#6802) 2021-04-07 12:25:13 +01:00
test_dtype_device_mixin.py [feat] Add better support for predict + ddp 2/3 (#7215) 2021-04-27 08:46:45 -04:00
test_imports.py fix import and typo in AMP (#4871) 2020-11-26 23:45:52 +01:00
test_memory.py Make move_metrics_to_cpu work recursively (#6007) 2021-02-16 21:52:42 +00:00
test_model_helpers.py Add `functools.wraps` support for `is_overridden` (#8296) 2021-07-06 10:40:54 +02:00
test_parsing.py add forgotten test in #7240 (#7283) 2021-05-03 23:56:30 +00:00
test_remote_filesystem.py Add tests for GCS filesystem (#7946) 2021-07-05 12:16:32 +00:00
test_seed.py Fixed `num_sanity_val_steps` affecting reproducibility of training data shuffling (#7014) 2021-04-27 09:51:39 +00:00
test_upgrade_checkpoint.py formatting tests1/n (#5843) 2021-02-06 08:22:10 -05:00
test_warnings.py Actually show deprecation warnings and their line level [2/2] (#8002) 2021-06-21 18:51:53 +02:00
test_xla_device_utils.py [TPU] update is_tpu_exists utils internal logic to rely on xmp.spawn (#6719) 2021-03-29 18:59:20 +01:00