Cache Cox Model instance during runtime

JelleAalbers · Aug 15, 2024 · bfa4394 · bfa4394
1 parent 8c3e5eb
commit bfa4394
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/wimprates/data/migdal/Cox/cox_wrapper.py b/wimprates/data/migdal/Cox/cox_wrapper.py
@@ -5,17 +5,18 @@
 The working directory is then reset.
 """
 
+from functools import lru_cache
 import os
 import sys
 
-import wimprates as wr
-
 from .cox_submodule.Migdal import Migdal
+import wimprates as wr
 
 export, __all__ = wr.exporter()
 
 
 @export
+@lru_cache
 def cox_migdal_model(element: str, **kwargs) -> Migdal:
     """
     This function creates a Cox Migdal model for a given element.