From 10bec87014c329f25b8d3dff74b70842fac1e1c0 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sat, 3 Aug 2019 10:08:51 -0500 Subject: [PATCH] fixed import --- pytorch_lightning/__init__.py | 2 +- setup.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/pytorch_lightning/__init__.py b/pytorch_lightning/__init__.py index e7a03ba8..4b5961ac 100644 --- a/pytorch_lightning/__init__.py +++ b/pytorch_lightning/__init__.py @@ -1,3 +1,3 @@ -from .models import Trainer +from .models.trainer import Trainer from .root_module.root_module import LightningModule from .root_module.decorators import data_loader \ No newline at end of file diff --git a/setup.py b/setup.py index 51658c2e..aed7c680 100755 --- a/setup.py +++ b/setup.py @@ -19,7 +19,7 @@ setup( install_requires=[ "torch>=1.1.0", "tqdm", - "test-tube==0.6.7.4", + "test-tube>=0.6.7.6", ], packages=find_packages(), long_description=open("README.md", encoding="utf-8").read(),