diff --git a/src/aiida_wannier90/calculations/postw90.py b/src/aiida_wannier90/calculations/postw90.py index 8d82025..de6c9d0 100644 --- a/src/aiida_wannier90/calculations/postw90.py +++ b/src/aiida_wannier90/calculations/postw90.py @@ -78,7 +78,7 @@ class Postw90Calculation(CalcJob): ".vdw", "_band_proj.dat", "_band.labelinfo.dat", - ".node_00001.werr", + ".node_*.werr", ) _DEFAULT_RETRIEVE_TEMPORARY_SUFFIXES = ( diff --git a/src/aiida_wannier90/calculations/wannier90.py b/src/aiida_wannier90/calculations/wannier90.py index 21c6213..4fd5313 100644 --- a/src/aiida_wannier90/calculations/wannier90.py +++ b/src/aiida_wannier90/calculations/wannier90.py @@ -122,7 +122,7 @@ class Wannier90Calculation(CalcJob): ".vdw", "_band_proj.dat", "_band.labelinfo.dat", - ".node_00001.werr", + ".node_*.werr", ) @classmethod diff --git a/src/aiida_wannier90/parsers/postw90.py b/src/aiida_wannier90/parsers/postw90.py index 40c5c56..24b196f 100644 --- a/src/aiida_wannier90/parsers/postw90.py +++ b/src/aiida_wannier90/parsers/postw90.py @@ -120,7 +120,8 @@ def parse(self, **kwargs): # pylint: disable=inconsistent-return-statements ) return self.exit_codes.ERROR_WERR_FILE_PRESENT - # Some times the error files are aiida.node_00001.werr, ... + # Some times the error files are aiida.node_XXXXX.werr, ... + # The XXXXX are 5-digit index of processor error_file_name = re.compile(seedname + r".+?\.werr") for filename in out_folder.base.repository.list_object_names(): if error_file_name.match(filename): diff --git a/src/aiida_wannier90/parsers/wannier90.py b/src/aiida_wannier90/parsers/wannier90.py index 4b2206d..7a18a4a 100644 --- a/src/aiida_wannier90/parsers/wannier90.py +++ b/src/aiida_wannier90/parsers/wannier90.py @@ -126,7 +126,8 @@ def parse(self, **kwargs): # pylint: disable=inconsistent-return-statements ) return self.exit_codes.ERROR_WERR_FILE_PRESENT - # Some times the error files are aiida.node_00001.werr, ... + # Some times the error files are aiida.node_XXXXX.werr, ... + # The XXXXX are 5-digit index of processor error_file_name = re.compile(seedname + r".+?\.werr") for filename in out_folder.base.repository.list_object_names(): if error_file_name.match(filename): diff --git a/tests/calculations/test_local_input.py b/tests/calculations/test_local_input.py index 43b6999..ef95557 100644 --- a/tests/calculations/test_local_input.py +++ b/tests/calculations/test_local_input.py @@ -105,7 +105,7 @@ def test_default( ".vdw", "_band_proj.dat", "_band.labelinfo.dat", - ".node_00001.werr", + ".node_*.werr", ) ] retrieve_temporary_list = [] @@ -191,7 +191,7 @@ def test_default_plot( ".vdw", "_band_proj.dat", "_band.labelinfo.dat", - ".node_00001.werr", + ".node_*.werr", ) ] retrieve_temporary_list = [] @@ -304,7 +304,7 @@ def test_no_projections( ".vdw", "_band_proj.dat", "_band.labelinfo.dat", - ".node_00001.werr", + ".node_*.werr", ) ] retrieve_temporary_list = [] @@ -369,7 +369,7 @@ def test_list_projections( ".vdw", "_band_proj.dat", "_band.labelinfo.dat", - ".node_00001.werr", + ".node_*.werr", ) ] retrieve_temporary_list = [] diff --git a/tests/calculations/test_remote_input.py b/tests/calculations/test_remote_input.py index 027bd43..37632c4 100644 --- a/tests/calculations/test_remote_input.py +++ b/tests/calculations/test_remote_input.py @@ -104,7 +104,7 @@ def test_default_remote( ".vdw", "_band_proj.dat", "_band.labelinfo.dat", - ".node_00001.werr", + ".node_*.werr", ) ] retrieve_temporary_list = [] @@ -201,7 +201,7 @@ def test_unk_symlink( ".vdw", "_band_proj.dat", "_band.labelinfo.dat", - ".node_00001.werr", + ".node_*.werr", ) ] retrieve_temporary_list = []