From c17d72841f9bc020f5aa417b63169396ceb9fc3f Mon Sep 17 00:00:00 2001
From: Konsta Vesterinen <konsta.vesterinen@gmail.com>
Date: Mon, 21 Apr 2014 11:13:07 +0300
Subject: [PATCH] Add get_columns to functions module main import

---
 sqlalchemy_utils/functions/__init__.py | 2 ++
 tests/functions/test_get_columns.py    | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/sqlalchemy_utils/functions/__init__.py b/sqlalchemy_utils/functions/__init__.py
index bc6caeb..a60d6ce 100644
--- a/sqlalchemy_utils/functions/__init__.py
+++ b/sqlalchemy_utils/functions/__init__.py
@@ -13,6 +13,7 @@ from .database import (
 )
 from .orm import (
     declarative_base,
+    get_columns,
     getdotattr,
     has_changes,
     identity,
@@ -30,6 +31,7 @@ __all__ = (
     'defer_except',
     'drop_database',
     'escape_like',
+    'get_columns',
     'getdotattr',
     'has_changes',
     'identity',
diff --git a/tests/functions/test_get_columns.py b/tests/functions/test_get_columns.py
index 270486a..4cb866e 100644
--- a/tests/functions/test_get_columns.py
+++ b/tests/functions/test_get_columns.py
@@ -1,5 +1,5 @@
 import sqlalchemy as sa
-from sqlalchemy_utils.functions.orm import get_columns
+from sqlalchemy_utils.functions import get_columns
 from tests import TestCase