aiida-v2 support: EntityExtrasMixin -> EntityExtras

reference: https://aiida.readthedocs.io/projects/aiida-core/en/latest/reference/_changelog.html#v2-0-0-2022-04-27 (also ran aiida-upgrade, did not find anything)
JuDFTteam · Dec 14, 2022 · 6fd468e · 6fd468e
1 parent 60b880c
commit 6fd468e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/aiida_jutools/meta/extra/forms/KkrConstantsVersion.py b/aiida_jutools/meta/extra/forms/KkrConstantsVersion.py
@@ -50,7 +50,7 @@ def __init__(self,
         }
 
     def load(self,
-             entity: _orm.EntityExtrasMixin,
+             entity: _orm.EntityExtras,
              **kwargs):
         """This methods extends :py:meth:`~.util.ExtraForm.load`.
 

diff --git a/aiida_jutools/meta/extra/util.py b/aiida_jutools/meta/extra/util.py
@@ -55,7 +55,7 @@ def value(self) -> _typing.Any:
 
     @abc.abstractmethod
     def load(self,
-             entity: _orm.EntityExtrasMixin,
+             entity: _orm.EntityExtras,
              **kwargs):
         """Load an entity's extra's value into this form.
 
@@ -82,7 +82,7 @@ def clear(self):
         self._value.pop(self._error_report_key, None)
 
     def insert(self,
-               entity: _orm.EntityExtrasMixin,
+               entity: _orm.EntityExtras,
                validate: bool = False,
                overwrite: bool = False,
                **kwargs) -> bool: