From 48f658fbb551e5f65a32938224dd782dd4605820 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 13 Aug 2020 21:44:55 -0400 Subject: [PATCH] Fixes #2943 (#2970) --- pytorch_lightning/core/datamodule.py | 2 +- pytorch_lightning/trainer/trainer.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/pytorch_lightning/core/datamodule.py b/pytorch_lightning/core/datamodule.py index 9e9f641409..657b35e9fe 100644 --- a/pytorch_lightning/core/datamodule.py +++ b/pytorch_lightning/core/datamodule.py @@ -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 ( diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py index 8b45d10986..7fc8b93b4d 100644 --- a/pytorch_lightning/trainer/trainer.py +++ b/pytorch_lightning/trainer/trainer.py @@ -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 (