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

Add _gpus_arg_default in argparse_utils for backward compatibility (#7402)

This commit is contained in:
Gyeongjae Choi 2021-05-06 22:35:12 +09:00 committed by GitHub
parent 94f6c3e160
commit d9bdc56b6a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pytorch_lightning/utilities/argparse_utils.py
View File

@ -3,3 +3,7 @@ from pytorch_lightning.utilities import rank_zero_deprecation
rank_zero_deprecation("`argparse_utils` package has been renamed to `argparse` since v1.2 and will be removed in v1.4")
from pytorch_lightning.utilities.argparse import * # noqa: F403 E402 F401
# for backward compatibility with old checkpoints (versions < 1.2.0)
# that need to be able to unpickle the function from the checkpoint
from pytorch_lightning.utilities.argparse import _gpus_arg_default # noqa: E402 F401 # isort: skip