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

Fixes #2943 (#2970)

This commit is contained in:
William Falcon 2020-08-13 21:44:55 -04:00 committed by GitHub
parent 639a4cbd25
commit 48f658fbb5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch_lightning/core/datamodule.py
View File

@ -317,7 +317,7 @@ class LightningDataModule(object, metaclass=_DataModuleWrapper): # pragma: no c
depr_arg_names = blacklist + added_args
depr_arg_names = set(depr_arg_names)
allowed_types = (str, float, int, bool)
allowed_types = (str, int, float, bool)
# TODO: get "help" from docstring :)
for arg, arg_types, arg_default in (

2
pytorch_lightning/trainer/trainer.py
View File

@ -736,7 +736,7 @@ class Trainer(
blacklist = ['kwargs']
depr_arg_names = cls.get_deprecated_arg_names() + blacklist
allowed_types = (str, float, int, bool)
allowed_types = (str, int, float, bool)
# TODO: get "help" from docstring :)
for arg, arg_types, arg_default in (