Cleaning

1 files changed, 1 insertions, 7 deletions

diff --git a/factory/alchemy.py b/factory/alchemy.py
index 87a17fa..8ede97f 100644
--- a/factory/alchemy.py
+++ b/factory/alchemy.py
@@ -18,15 +18,8 @@
 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 # THE SOFTWARE.
-
 from . import base
 
-try:
-    from sqlalchemy.sql.functions import max
-except ImportError:
-    def max():
-        raise ImportError('sqlalchemy is not found!')
-
 
 class SQLAlchemyModelFactory(base.Factory):
     """Factory for SQLAlchemy models. """
@@ -36,6 +29,7 @@ class SQLAlchemyModelFactory(base.Factory):
     @classmethod
     def _setup_next_sequence(cls, *args, **kwargs):
         """Compute the next available PK, based on the 'pk' database field."""
+        from sqlalchemy.sql.functions import max
         session = cls.FACTORY_SESSION
         pk = cls.FACTORY_FOR.__table__.primary_key.columns.values()[0].key
         max_pk = session.query(max(getattr(cls.FACTORY_FOR, pk))).one()