remove mentions to flat arrays in docstrings

mcx/inference/adaptive.py

@@ -8,11 +8,6 @@
 The Stan Manual [1]_ is a very good reference on automatic tuning of
 parameters used in Hamiltonian Monte Carlo.
 
-.. note:
-    This is a "flat zone": values used to update the step size or the mass
-    matrix are 1D arrays. Raveling/unraveling logic should happen at a higher
-    level.
-
 .. [1]: "HMC Algorithm Parameters", Stan Manual
         https://mc-stan.org/docs/2_20/reference-manual/hmc-algorithm-parameters.html
 """

mcx/inference/integrators.py

@@ -2,10 +2,6 @@
 manifolds.
 
 IntegratorStep does not need to compute and return the logprob!
-
-.. note:
-    This file is a "flat zone": position and momentum are one-dimensional arrays.
-    Any raveling/unraveling logic must be placed at a higher level.
 """
 from functools import partial
 from typing import Callable, NamedTuple

mcx/inference/kernels.py

@@ -1,8 +1,4 @@
 """Sampling kernels.
-
-.. note:
-    This file is a "flat zone": positions and logprobs are 1 dimensional
-    arrays. Raveling and unraveling logic must happen outside.
 """
 from typing import Callable, NamedTuple, Tuple
 

mcx/inference/metrics.py

@@ -1,8 +1,4 @@
 """Generate dynamics on a Euclidean Manifold or Riemannian Manifold.
-
-.. note:
-    This file is a "flat zone": positions and logprobs are 1 dimensional
-    arrays. Raveling and unraveling logic must happen outside.
 """
 from typing import Callable, Tuple
 

mcx/inference/warmup.py

@@ -1,7 +1,4 @@
 """Warming up the chain.
-
-.. note:
-    This is a "flat zone": all positions are 1D array.
 """
 from functools import partial
 from typing import Callable, Tuple