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