From 34b5f7ae6d2377332d12035fa817de9f9504c6ee 2009-05-16 18:08:21
From: Pauli Virtanen <pauli.virtanen@iki.fi>
Date: 2009-05-16 18:08:21
Subject: [PATCH] ipy_autoreload: fix %aimporting submodules
---

diff --git a/IPython/Extensions/ipy_autoreload.py b/IPython/Extensions/ipy_autoreload.py
index e64161d..ad40483 100644
--- a/IPython/Extensions/ipy_autoreload.py
+++ b/IPython/Extensions/ipy_autoreload.py
@@ -328,9 +328,12 @@ def aimport_f(self, parameter_s=''):
         try: del reloader.skip_modules[modname]
         except KeyError: pass
         reloader.modules[modname] = True
-
-        mod = __import__(modname)
-        ip.to_user_ns({modname: mod})
+        
+        # Inject module to user namespace; handle also submodules properly
+        __import__(modname)
+        basename = modname.split('.')[0]
+        mod = sys.modules[basename]
+        ip.to_user_ns({basename: mod})
 
 def init():
     ip.expose_magic('autoreload', autoreload_f)