Fix typing

Author: edan-bainglass

src/aiida/cmdline/commands/cmd_code.py

@@ -41,7 +41,7 @@ def create_code(ctx: click.Context, cls: Code, **kwargs) -> None:
     """Create a new `Code` instance."""
     try:
         model = cls.InputModel(**kwargs)
-        instance = cls.from_model(model)  # type: ignore[arg-type]
+        instance = cls.from_model(model)
     except (TypeError, ValueError) as exception:
         echo.echo_critical(f'Failed to create instance `{cls}`: {exception}')
 

src/aiida/orm/nodes/__init__.py

@@ -43,7 +43,9 @@
     'List',
     'Node',
     'NodeAttributes',
+    'NodeModelType',
     'NodeRepository',
+    'NodeType',
     'NumericType',
     'OrbitalData',
     'PortableCode',

src/aiida/orm/nodes/data/bool.py

@@ -13,6 +13,7 @@
 from aiida.common.pydantic import MetadataField
 
 from . import base
+from .base import to_aiida_type
 
 __all__ = ('Bool',)
 
@@ -38,8 +39,8 @@ def __bool__(self):
         return self.value
 
 
-@base.to_aiida_type.register(bool)
-@base.to_aiida_type.register(numpy.bool_)
+@to_aiida_type.register(bool)
+@to_aiida_type.register(numpy.bool_)
 def _(value):
     return Bool(value)
 

src/aiida/orm/nodes/data/str.py

@@ -11,6 +11,7 @@
 from aiida.common.pydantic import MetadataField
 
 from . import base
+from .base import to_aiida_type
 
 __all__ = ('Str',)
 
@@ -30,6 +31,6 @@ class Str(base.BaseType):
     _type = str
 
 
-@base.to_aiida_type.register(str)
+@to_aiida_type.register(str)
 def _(value):
     return Str(value)

src/aiida/orm/nodes/node.py

@@ -64,7 +64,11 @@
     from ..implementation.nodes import BackendNode
     from .repository import NodeRepository
 
-__all__ = ('Node',)
+__all__ = (
+    'Node',
+    'NodeModelType',
+    'NodeType',
+)
 
 NodeType = TypeVar('NodeType', bound='Node')
 NodeModelType = TypeVar('NodeModelType', bound='Model')
@@ -374,7 +378,7 @@ def __init__(
             self.base.extras.set_many(extras)
 
     @classmethod
-    def from_model(cls, model: Model) -> Self:  # type: ignore[override]
+    def from_model(cls, model: NodeModelType) -> Self:
         """Return an entity instance from an instance of its model."""
         fields = cls.model_to_orm_field_values(model)