# Path: blackjax/types.py



from typing import Callable, NamedTuple, Tuple
from blackjax.types import PyTree
import jax
import jax.numpy as jnp




class SMCInfo(NamedTuple):
    """Additional information on the tempered SMC step.

    weights: jnp.ndarray
        The weights after the MCMC pass.
    proposals: PyTree
        The particles that were proposed by the MCMC pass.
    ancestors: jnp.ndarray
        The index of the particles proposed by the MCMC pass that were selected
        by the resampling step.
    log_likelihood_increment: float
        The log-likelihood increment due to the current step of the SMC algorithm.
    """

    weights: jnp.ndarray

# Path: blackjax/types.py



# Path: blackjax/smc/base.py
from typing import Callable, NamedTuple, Tuple
from blackjax.types import PyTree
import jax
import jax.numpy as jnp





class SMCInfo(NamedTuple):
    """Additional information on the tempered SMC step.

    weights: jnp.ndarray
        The weights after the MCMC pass.
    proposals: PyTree
        The particles that were proposed by the MCMC pass.
    ancestors: jnp.ndarray
        The index of the particles proposed by the MCMC pass that were selected
        by the resampling step.
    log_likelihood_increment: float
        The log-likelihood increment due to the current step of the SMC algorithm.
    """

    weights: jnp.ndarray

    proposals: PyTree