Fixing mispelt references in docstrings

mici/samplers.py

@@ -968,7 +968,7 @@ def sample_chains_with_adaptive_warm_up(
 
         Args:
             n_warm_up_iter (int): Number of adaptive warm up iterations per
-                chain. Depending on the `mici.stages.Stager` instance specified
+                chain. Depending on the `mici.stagers.Stager` instance specified
                 by the `stage arguments the warm up iterations may be split
                 between one or more adaptive stages.
             n_main_iter (int): Number of iterations (samples to draw) per chain
@@ -1431,7 +1431,7 @@ def sample_chains_with_adaptive_warm_up(
 
         Args:
             n_warm_up_iter (int): Number of adaptive warm up iterations per
-                chain. Depending on the `mici.stages.Stager` instance specified
+                chain. Depending on the `mici.stagers.Stager` instance specified
                 by the `stage arguments the warm up iterations may be split
                 between one or more adaptive stages.
             n_main_iter (int): Number of iterations (samples to draw) per chain

mici/states.py

@@ -16,16 +16,16 @@ def _cache_key_func(system, method):
 def cache_in_state(*depends_on):
     """Memoizing decorator for system methods.
 
-    Used to decorate `mici.system.System` methods which compute a function of
+    Used to decorate `mici.systems.System` methods which compute a function of
     one or more chain state variable(s), with the decorated method caching the
     value returned by the method being wrapped in the `ChainState` object to
     prevent the need for recomputation on future calls if the state variables
     the returned value depends on have not been changed in between the calls.
 
     Additionally for `ChainState` instances initialized with a `_call_counts`
     argument, the memoized method will update a counter for the method in the
-    `ChainState._call_counts` attribute every time the method being decorated is
-    called (i.e. when there isn't a valid cached value available).
+    `_call_counts` attribute every time the method being decorated is called
+    (i.e. when there isn't a valid cached value available).
 
     Args:
        *depends_on: One or more strings corresponding to the names of any state
@@ -53,7 +53,7 @@ def wrapper(self, state):
 def cache_in_state_with_aux(depends_on, auxiliary_outputs):
     """Memoizing decorator for system methods with possible auxiliary outputs.
 
-    Used to decorate `mici.system.System` methods which compute a function of
+    Used to decorate `mici.systems.System` methods which compute a function of
     one or more chain state variable(s), with the decorated method caching the
     value or values returned by the method being wrapped in the `ChainState`
     object to prevent the need for recomputation on future calls if the state
@@ -80,8 +80,8 @@ def cache_in_state_with_aux(depends_on, auxiliary_outputs):
 
     Additionally for `ChainState` instances initialized with a `_call_counts`
     argument, the memoized method will update a counter for the method in the
-    `ChainState._call_counts` attribute every time the method being decorated is
-    called (i.e. when there isn't a valid cached value available).
+    `_call_counts` attribute every time the method being decorated is called
+    (i.e. when there isn't a valid cached value available).
 
     Args:
         depends_on (str or Tuple[str]): A string or tuple of strings, with each