Fix typing for GraphQL types attributes

backend/infrahub/core/registry.py

@@ -35,7 +35,7 @@ class Registry:
     _default_branch: Optional[str] = None
     _default_ipnamespace: Optional[str] = None
     _schema: Optional[SchemaManager] = None
-    default_graphql_type: dict[str, InfrahubObject] = field(default_factory=dict)
+    default_graphql_type: dict[str, InfrahubObject | type[BaseAttribute]] = field(default_factory=dict)
     graphql_type: dict = field(default_factory=lambda: defaultdict(dict))
     data_type: dict[str, type[InfrahubDataType]] = field(default_factory=dict)
     input_type: dict[str, Union[BaseAttributeCreate, BaseAttributeUpdate]] = field(default_factory=dict)

backend/infrahub/graphql/types/attribute.py

@@ -1,5 +1,7 @@
 from __future__ import annotations
 
+from typing import Any
+
 from graphene import BigInt, Boolean, DateTime, Field, InputObjectType, Int, List, ObjectType, String
 from graphene.types.generic import GenericScalar
 
@@ -70,7 +72,7 @@ class BaseAttribute(ObjectType):
     is_from_profile = Field(Boolean)
 
     @classmethod
-    def __init_subclass__(cls, **kwargs):
+    def __init_subclass__(cls, **kwargs: dict[str, Any]) -> None:
         super().__init_subclass__(**kwargs)
         registry.default_graphql_type[cls.__name__] = cls
 

pyproject.toml

@@ -329,11 +329,6 @@ ignore_errors = true
 module = "infrahub.graphql.subscription"
 ignore_errors = true
 
-
-[[tool.mypy.overrides]]
-module = "infrahub.graphql.types.attribute"
-ignore_errors = true
-
 [[tool.mypy.overrides]]
 module = "infrahub.graphql.types.standard_node"
 ignore_errors = true