From 226b27fc4e5870229964375f26dff78f8aaa76b5 Mon Sep 17 00:00:00 2001 From: Markus Padourek Date: Tue, 7 Jun 2016 10:07:49 +0100 Subject: [PATCH] Allow node interface to be used in ConnectionField --- graphene/relay/fields.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/graphene/relay/fields.py b/graphene/relay/fields.py index 9bd097ec..65ecb583 100644 --- a/graphene/relay/fields.py +++ b/graphene/relay/fields.py @@ -58,10 +58,10 @@ class ConnectionField(Field): return edge_type.for_node(node) def get_type(self, schema): - from graphene.relay.utils import is_node + from graphene.relay.utils import is_node, is_node_type type = schema.T(self.type) node = schema.objecttype(type) - assert is_node(node), 'Only nodes have connections.' + assert is_node(node) or is_node_type(node), 'Only nodes have connections.' schema.register(node) connection_type = self.get_connection_type(node)