From 2870259c5cce9eb1cc22bd44a18e149e61ba3c65 Mon Sep 17 00:00:00 2001
From: Crivella <davidecrivella@yahoo.it>
Date: Tue, 22 Nov 2022 11:56:44 +0100
Subject: [PATCH] Fixed more deprecation warnins

---
 aiida_z2pack/calculations/z2pack.py | 10 +++++-----
 pyproject.toml                      |  2 +-
 tests/conftest.py                   |  2 +-
 3 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/aiida_z2pack/calculations/z2pack.py b/aiida_z2pack/calculations/z2pack.py
index 9a3a33b..26b0439 100644
--- 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
index bc5c537..7dc0875 100644
--- 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
index 497ffae..f0a22d6 100644
--- 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):