From e3809e9aee88b12358f5e2e290593f0b1b3f7ef3 Mon Sep 17 00:00:00 2001 From: Erik Wrede Date: Fri, 3 Mar 2023 11:30:09 +0100 Subject: [PATCH] chore: streamline edge creation --- graphene/relay/connection.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/graphene/relay/connection.py b/graphene/relay/connection.py index 1a41c588..ad7a7138 100644 --- a/graphene/relay/connection.py +++ b/graphene/relay/connection.py @@ -25,10 +25,10 @@ def get_edge_class( description = f"A Relay edge containing a `{base_name}` and its cursor." edge_name = f"{base_name}Edge" - if edge_class: - edge_bases = [edge_class, EdgeBase, ObjectType] - else: - edge_bases = [EdgeBase, ObjectType] + + edge_bases = [edge_class] if edge_class else [EdgeBase] + if not isinstance(edge_class, ObjectType): + edge_bases = [*edge_bases, ObjectType] return type(edge_name, tuple(edge_bases), {"Meta": EdgeMeta})