Add in memory matrix cache

ecmwf · Oct 8, 2024 · dae034e · dae034e
1 parent 3c32d3e
commit dae034e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/earthkit/regrid/utils/memcache.py b/src/earthkit/regrid/utils/memcache.py
@@ -77,7 +77,7 @@ class NoPolicy(MemoryCachePolicy):
     def check(self):
         self.cache.max_mem = 0
         self.cache.pre_check_matrix_size = False
-        self.ensure_capacity = False
+        self.cache.ensure_capacity = False
 
     def reduce(self, *args, **kwargs):
         self.cache._clear()
@@ -95,7 +95,7 @@ class UnlimitedPolicy(MemoryCachePolicy):
     def check(self):
         self.cache.max_mem = None
         self.cache.pre_check_matrix_size = False
-        self.ensure_capacity = False
+        self.cache.ensure_capacity = False
 
     def reduce(self, *args, **kwargs):
         # must be called within a lock