diff --git a/graphene/__init__.py b/graphene/__init__.py index 089d4765..5cbd2ab9 100644 --- a/graphene/__init__.py +++ b/graphene/__init__.py @@ -30,7 +30,8 @@ if not __SETUP__: List, NonNull, Enum, Argument, - Dynamic + Dynamic, + Union, ) from .relay import ( Node, @@ -62,6 +63,7 @@ if not __SETUP__: 'NonNull', 'Argument', 'Dynamic', + 'Union', 'resolve_only_args', 'Node', 'is_node', diff --git a/graphene/types/__init__.py b/graphene/types/__init__.py index c7411ff9..179c6f6d 100644 --- a/graphene/types/__init__.py +++ b/graphene/types/__init__.py @@ -13,6 +13,7 @@ from .inputfield import InputField from .argument import Argument from .inputobjecttype import InputObjectType from .dynamic import Dynamic +from .union import Union __all__ = [ @@ -33,5 +34,6 @@ __all__ = [ 'Boolean', 'List', 'NonNull', - 'Argument' - 'Dynamic'] + 'Argument', + 'Dynamic', +]