Cache the loss of a SequenceLearner (#411)

python-adaptive · Apr 16, 2023 · 541e1f2 · 541e1f2
1 parent 295516b
commit 541e1f2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/adaptive/learner/sequence_learner.py b/adaptive/learner/sequence_learner.py
@@ -8,7 +8,11 @@
 from sortedcontainers import SortedDict, SortedSet
 
 from adaptive.learner.base_learner import BaseLearner
-from adaptive.utils import assign_defaults, partial_function_from_dataframe
+from adaptive.utils import (
+    assign_defaults,
+    cache_latest,
+    partial_function_from_dataframe,
+)
 
 try:
     import pandas
@@ -113,6 +117,7 @@ def ask(
 
         return points, loss_improvements
 
+    @cache_latest
     def loss(self, real: bool = True) -> float:
         if not (self._to_do_indices or self.pending_points):
             return 0.0

diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -17,7 +17,7 @@
 # -- Project information -----------------------------------------------------
 
 project = "adaptive"
-copyright = "2018-2022, Adaptive Authors"
+copyright = "2018-2023, Adaptive Authors"
 author = "Adaptive Authors"
 
 # The short X.Y version