From b993a3ed399de2fdf2d5fd9f5646160d53fa2588 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 27 Apr 2020 07:21:58 -0400 Subject: [PATCH] ddp pickle --- tests/trainer/test_trainer_cli.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tests/trainer/test_trainer_cli.py b/tests/trainer/test_trainer_cli.py index bfc67111a7..93cbb8e2cc 100644 --- a/tests/trainer/test_trainer_cli.py +++ b/tests/trainer/test_trainer_cli.py @@ -47,6 +47,11 @@ def test_add_argparse_args_redefined(cli_args): assert depr_name not in args trainer = Trainer.from_argparse_args(args=args) + + # make sure trainer can be pickled + import pickle + pickle.dumps(trainer) + assert isinstance(trainer, Trainer)