Fixed more deprecation warnins

AntimoMarrazzo · Nov 22, 2022 · 2870259 · 2870259
1 parent b425e29
commit 2870259
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/aiida_z2pack/calculations/z2pack.py b/aiida_z2pack/calculations/z2pack.py
@@ -112,22 +112,22 @@ def define(cls, spec):
             )
 
         spec.input(
-            'pw_code', valid_type=orm.Code,
+            'pw_code', valid_type=orm.AbstractCode,
             required=False,
             help='NSCF code to be used by z2pack.'
             )
         spec.input(
-            'overlap_code', valid_type=orm.Code,
+            'overlap_code', valid_type=orm.AbstractCode,
             required=False,
             help='Overlap code to be used by z2pack.'
             )
         spec.input(
-            'wannier90_code', valid_type=orm.Code,
+            'wannier90_code', valid_type=orm.AbstractCode,
             required=False,
             help='Wannier code to be used by z2pack.'
             )
         spec.input(
-            'code', valid_type=orm.Code,
+            'code', valid_type=orm.AbstractCode,
             required=False,
             help='Z2pack code.'
             )
@@ -285,7 +285,7 @@ def _set_inputs_from_parent_scf(self):
         parent = self.inputs.parent_folder
         calc   = parent.creator
 
-        pseudos      = calc.get_incoming(link_label_filter='pseudos%').all()
+        pseudos      = calc.base.links.get_incoming(link_label_filter='pseudos%').all()
         pseudos_dict = {name[9:]:upf for upf,_,name in pseudos}
         self.inputs.pseudos       = pseudos_dict
         self.inputs.structure     = calc.inputs.structure

diff --git a/pyproject.toml b/pyproject.toml
@@ -28,7 +28,7 @@ dependencies = [
     'z2pack==2.1.1',
     'aiida-core~=2.1.2',
     'aiida_quantumespresso~=4.0',
-    'aiida_wannier90~=2.0.0'
+    'aiida_wannier90~=3.0.0'
 ]
 
 [project.urls]

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -114,7 +114,7 @@ def _generate_calc_job_node(entry_point_name, computer, test_name=None, inputs=N
         node.set_option('max_wallclock_seconds', 1800)
 
         if attributes:
-            node.set_attribute_many(attributes)
+            node.base.attributes.set_many(attributes)
 
         if inputs:
             for link_label, input_node in flatten_inputs(inputs):