Add cached class to other compat. schemes

materialsproject · Mar 2, 2022 · 2ad326e · 2ad326e
1 parent 4b66ebc
commit 2ad326e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pymatgen/entries/compatibility.py b/pymatgen/entries/compatibility.py
@@ -817,7 +817,7 @@ def __init__(self, compat_type="Advanced", correct_peroxide=True, check_potcar_h
             ]
         )
 
-
+@cached_class
 class MaterialsProject2020Compatibility(Compatibility):
     """
     This class implements the Materials Project 2020 energy correction scheme,
@@ -1177,7 +1177,7 @@ def __init__(self, compat_type="Advanced", correct_peroxide=True, check_potcar_h
             ]
         )
 
-
+@cached_class
 class MaterialsProjectAqueousCompatibility(Compatibility):
     """
     This class implements the Aqueous energy referencing scheme for constructing