Use dict.clear() to clear Parameter cache

loganbvh · Apr 12, 2024 · 2bd0830 · 2bd0830
1 parent 74ccf26
commit 2bd0830
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tdgl/solver/solver.py b/tdgl/solver/solver.py
@@ -216,9 +216,9 @@ def disorder_epsilon(r):
 
         # Clear the Parameter caches
         if isinstance(self.applied_vector_potential, Parameter):
-            self.applied_vector_potential._cache = {}
+            self.applied_vector_potential._cache.clear()
         if isinstance(self.disorder_epsilon, Parameter):
-            self.disorder_epsilon._cache = {}
+            self.disorder_epsilon._cache.clear()
 
         # Find the current terminal sites.
         self.terminal_info = device.terminal_info()
@@ -807,9 +807,9 @@ def solve(self) -> Optional[Solution]:
 
             # Clear the Parameter caches
             if isinstance(self.applied_vector_potential, Parameter):
-                self.applied_vector_potential._cache = {}
+                self.applied_vector_potential._cache.clear()
             if isinstance(self.disorder_epsilon, Parameter):
-                self.disorder_epsilon._cache = {}
+                self.disorder_epsilon._cache.clear()
 
             solution = None
             if data_was_generated: