gwkokab.analysis.core.numpyro_base
==================================

.. py:module:: gwkokab.analysis.core.numpyro_base


Attributes
----------

.. autoapisummary::

   gwkokab.analysis.core.numpyro_base.numpyro_arg_parser


Classes
-------

.. autoapisummary::

   gwkokab.analysis.core.numpyro_base.NumpyroBase


Module Contents
---------------

.. py:class:: NumpyroBase(*, analysis_name: str, check_leaks: bool, debug_nans: bool, model: Union[numpyro.distributions.distribution.Distribution, collections.abc.Callable[Ellipsis, numpyro.distributions.distribution.Distribution]], poisson_mean_filename: str, prior_filename: str, profile_memory: bool, sampler_cfg, variance_cut_threshold: float | None)

   Bases: :py:obj:`gwkokab.analysis.core.analysis_base.AnalysisBase`


   AnalysisBase is a class which contains all the common functionality among the
   different analyses.

   It is not meant to be used directly, but rather to be subclassed by the specific
   analyses.


   .. py:method:: driver(*, logpdf: collections.abc.Callable[[jaxtyping.Array, Dict[str, Any]], jaxtyping.Array], priors: gwkokab.models.utils.JointDistribution, data: Dict[str, Any], labels: List[str]) -> None


.. py:data:: numpyro_arg_parser