ASE: not all calculators implement the get_stress() function

pyiron · Feb 7, 2024 · b54495e · b54495e
1 parent 0281e22
commit b54495e
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/atomistics/calculators/ase.py b/atomistics/calculators/ase.py
@@ -5,6 +5,7 @@
 from ase.md.npt import NPT
 from ase.md.velocitydistribution import MaxwellBoltzmannDistribution
 from ase.constraints import UnitCellFilter
+from ase.calculators.calculator import PropertyNotImplementedError
 import numpy as np
 from typing import TYPE_CHECKING
 
@@ -45,10 +46,16 @@ def energy_tot(self):
         )
 
     def stress(self):
-        return self.structure.get_stress(voigt=False)
+        try:
+            return self.structure.get_stress(voigt=False)
+        except PropertyNotImplementedError:
+            return None
 
     def pressure(self):
-        return self.structure.get_stress(voigt=False)
+        try:
+            return self.structure.get_stress(voigt=False)
+        except PropertyNotImplementedError:
+            return None
 
     def cell(self):
         return self.structure.get_cell()