fixed import

2019-08-03 10:08:51 -05:00 · 2019-08-03 10:08:51 -05:00 · 10bec87014
parent 6bccc38294
commit 10bec87014
2 changed files with 2 additions and 2 deletions
--- 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
--- 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(),