From d27ff46920da80bcb38d8fc3d6e68e13250faa07 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 16 Jan 2020 16:33:58 -0500 Subject: [PATCH] cleared spaces --- pytorch_lightning/core/decorators.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pytorch_lightning/core/decorators.py b/pytorch_lightning/core/decorators.py index 0a87e00f..aeea1a7e 100644 --- a/pytorch_lightning/core/decorators.py +++ b/pytorch_lightning/core/decorators.py @@ -1,4 +1,5 @@ import traceback +from functools import wraps def data_loader(fn): @@ -8,6 +9,7 @@ def data_loader(fn): :return: """ + wraps(fn) attr_name = '_lazy_' + fn.__name__ def _get_data_loader(self):