From ced4296897f11bd5e33768172fcad7aa2ef32f67 Mon Sep 17 00:00:00 2001 From: Sanghun Lee Date: Mon, 4 Dec 2023 17:28:41 +0900 Subject: [PATCH] chore: change type name of AsyncNode to Node --- src/ai/backend/manager/models/etcd.py | 3 ++- src/ai/backend/manager/models/gql_relay.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/ai/backend/manager/models/etcd.py b/src/ai/backend/manager/models/etcd.py index 3c1b9d3afe..8633d56f60 100644 --- a/src/ai/backend/manager/models/etcd.py +++ b/src/ai/backend/manager/models/etcd.py @@ -10,6 +10,7 @@ from ..defs import PASSWORD_PLACEHOLDER from . import UserRole from .base import privileged_mutation, set_if_set +from .gql_relay import AsyncNode if TYPE_CHECKING: from .gql import GraphQueryContext @@ -58,7 +59,7 @@ class ContainerRegistry(graphene.ObjectType): config = graphene.Field(ContainerRegistryConfig) class Meta: - interfaces = (graphene.relay.Node,) + interfaces = (AsyncNode,) # TODO: `get_node()` should be implemented to query a scalar object directly by ID # (https://docs.graphene-python.org/en/latest/relay/nodes/#nodes) diff --git a/src/ai/backend/manager/models/gql_relay.py b/src/ai/backend/manager/models/gql_relay.py index f8bee5eba5..63ab14cce9 100644 --- a/src/ai/backend/manager/models/gql_relay.py +++ b/src/ai/backend/manager/models/gql_relay.py @@ -81,7 +81,7 @@ class AsyncNode(Node): """ class Meta: - name = "AsyncNode" + name = "Node" @classmethod def Field(cls, *args, **kwargs):