Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

ddp pickle

This commit is contained in:
William Falcon 2020-04-27 07:44:45 -04:00
parent 5c0118fe9d
commit a3cebb4469
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/trainer/test_trainer_cli.py
View File

@ -1,6 +1,7 @@
import inspect import inspect
from argparse import ArgumentParser, Namespace from argparse import ArgumentParser, Namespace
from unittest import mock from unittest import mock
import pickle
import pytest import pytest
@ -42,14 +43,14 @@ def test_add_argparse_args_redefined(cli_args):
args = parser.parse_args(cli_args) args = parser.parse_args(cli_args)
# make sure we can pickle args
pickle.dumps(args)
# Check few deprecated args are not in namespace: # Check few deprecated args are not in namespace:
for depr_name in ('gradient_clip', 'nb_gpu_nodes', 'max_nb_epochs'): for depr_name in ('gradient_clip', 'nb_gpu_nodes', 'max_nb_epochs'):
assert depr_name not in args assert depr_name not in args
trainer = Trainer.from_argparse_args(args=args) trainer = Trainer.from_argparse_args(args=args)
# make sure trainer can be pickled
import pickle
pickle.dumps(trainer) pickle.dumps(trainer)
assert isinstance(trainer, Trainer) assert isinstance(trainer, Trainer)