Improved if/else statement

tilde-lab · Aug 20, 2024 · 9d7eb9a · 9d7eb9a
1 parent b19ec40
commit 9d7eb9a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/aiida_crystal_dft/parsers/cry_pycrystal.py b/aiida_crystal_dft/parsers/cry_pycrystal.py
@@ -169,9 +169,9 @@ def parse_out_trajectory(self, _):
         except ValueError as e:
             # Fix for calculation with SCELPHO keyword;
             # Scince supercell have more atoms than regular cell;
-            # traj.set_structurelist(structs) will throw an error https://github.com/aiidateam/aiida-core/blob/71422eb872040a9ba23047d2ec031f6deaa6a7cc/src/aiida/orm/nodes/data/array/trajectory.py#L202 
+            # traj.set_structurelist(structs) will throw an error https://github.com/aiidateam/aiida-core/blob/71422eb872040a9ba23047d2ec031f6deaa6a7cc/src/aiida/orm/nodes/data/array/trajectory.py#L202
             # There are no reason for tracking trajectory in phonon calculation, so it will return None
-            if "Phonon" in self._node.label:
+            if self.stdout_parser.info['phonons']['modes']:
                 self._logger.warning(f"Caught ValueError for node with label '{self._node.label}': {e}")
                 return None
             else: