install_mod can take strings or module items

diff --git a/lib/sqlalchemy/mods/__init__.py b/lib/sqlalchemy/mods/__init__.py
index 81650389acee935af7b5c220ebf10cff306603f8..328df3c561c476d135acde25a1b1f3908594b357 100644 (file)
--- a/lib/sqlalchemy/mods/__init__.py
+++ b/lib/sqlalchemy/mods/__init__.py
@@ -1,4 +1,7 @@
 def install_mods(*mods):
     for mod in mods:
-        _mod = getattr(__import__('sqlalchemy.mods.%s' % mod).mods, mod)
-        _mod.install_plugin()
\ No newline at end of file
+        if isinstance(mod, str):
+            _mod = getattr(__import__('sqlalchemy.mods.%s' % mod).mods, mod)
+            _mod.install_plugin()
+        else:
+            mod.install_plugin()
\ No newline at end of file