Merge pull request #283 from AlvaroGI/master

AverageLearner1D added

Author: basnijholt

README.rst

@@ -47,8 +47,10 @@ The following learners are implemented:
 - ``Learner1D``, for 1D functions ``f: ℝ → ℝ^N``,
 - ``Learner2D``, for 2D functions ``f: ℝ^2 → ℝ^N``,
 - ``LearnerND``, for ND functions ``f: ℝ^N → ℝ^M``,
-- ``AverageLearner``, For stochastic functions where you want to
+- ``AverageLearner``, for random variables where you want to
   average the result over many evaluations,
+- ``AverageLearner1D``, for stochastic 1D functions where you want to
+  estimate the mean value of the function at each point,
 - ``IntegratorLearner``, for
   when you want to intergrate a 1D function ``f: ℝ → ℝ``,
 - ``BalancingLearner``, for when you want to run several learners at once,

adaptive/__init__.py

@@ -4,6 +4,7 @@
 from adaptive._version import __version__
 from adaptive.learner import (
     AverageLearner,
+    AverageLearner1D,
     BalancingLearner,
     BaseLearner,
     DataSaver,
@@ -34,6 +35,7 @@
     "Learner1D",
     "Learner2D",
     "LearnerND",
+    "AverageLearner1D",
     "make_datasaver",
     "SequenceLearner",
     "active_plotting_tasks",

adaptive/learner/__init__.py

@@ -1,6 +1,7 @@
 from contextlib import suppress
 
 from adaptive.learner.average_learner import AverageLearner
+from adaptive.learner.average_learner1D import AverageLearner1D
 from adaptive.learner.balancing_learner import BalancingLearner
 from adaptive.learner.base_learner import BaseLearner
 from adaptive.learner.data_saver import DataSaver, make_datasaver
@@ -20,6 +21,7 @@
     "Learner1D",
     "Learner2D",
     "LearnerND",
+    "AverageLearner1D",
     "SequenceLearner",
 ]