lightning/tests/utilities
SpontaneousDuck dcff5036a8
Use PickleError base class to detect all pickle errors (#6917)
* Use PickleError base class to detect all pickle errors

* Update changelog with #6917

* Add pickle test for torch ScriptModule

Co-authored-by: Ken Witham <k.witham@kri.neu.edu>
Co-authored-by: Akihiro Nitta <nitta@akihironitta.com>
2021-04-14 20:24:32 +00:00
..
__init__.py New metric classes (#1326) (#1877) 2020-05-19 11:05:07 -04:00
distributed.py Fixed PYTHONPATH for ddp test model (#4528) 2020-12-05 20:09:47 +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 formatting tests1/n (#5843) 2021-02-06 08:22:10 -05:00
test_apply_func_torchtext.py fix importing torchtext batch (#6365) 2021-03-05 21:39:52 +01:00
test_argparse.py fix gpus default for Trainer.add_argparse_args (#6898) 2021-04-09 11:20:43 +02:00
test_cli.py Simple reproducibility with minimum boilerplate CLI training with `LightningCLI` (#4492) 2021-04-06 14:19:11 +01: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 Refactor: skipif for Windows 2/n (#6268) 2021-03-02 09:36:01 +00: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_parsing.py Use PickleError base class to detect all pickle errors (#6917) 2021-04-14 20:24:32 +00:00
test_seed.py fix imports / isort / flake8 2021-01-26 14:57:34 +01:00
test_upgrade_checkpoint.py formatting tests1/n (#5843) 2021-02-06 08:22:10 -05: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