From ae81473464523911688dd26b5d28fc350570fdee Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Wed, 3 Jul 2019 18:43:13 -0400
Subject: [PATCH] easy import for lightningModule

---
 pytorch_lightning/__init__.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/pytorch_lightning/__init__.py b/pytorch_lightning/__init__.py
index ce4b32be..5893eb4a 100644
--- a/pytorch_lightning/__init__.py
+++ b/pytorch_lightning/__init__.py
@@ -1 +1,2 @@
-from .models import Trainer
\ No newline at end of file
+from .models import Trainer
+from .root_module.root_module import LightningModule
\ No newline at end of file