def apply_impacts(
    baseline: tf.Tensor,
    multiplicative_impacts: list[tf.Tensor] | tuple[tf.Tensor, ...],
    additive_impacts: list[tf.Tensor] | tuple[tf.Tensor, ...],
) -> tf.Tensor:
    """Apply the impacts on top of the basline to get yhat.

    $$
    yhat = (baseline*\\underset{}{\\overset{m}{\\prod }}({multiplicativeEffect}_{m}))
        + \\underset{}{\\overset{a}{\\sum }}({additiveEffect}_{a})
    $$

    Args:
        baseline (tf.Tensor): The baseline impact. This will be the starting point where impacts are applied on.
        multiplicative_impacts (list[tf.Tensor]): These impacts scales the baseline multiplicatively (larger effect).
        additive_impacts (list[tf.Tensor]): These impacts, increase the baseline additively (smaller effect).

    Returns:
        tf.Tensor: The yhat after applying all the impacts on the baseline.
    """  # noqa: E501
    yhat = baseline
    for impact in multiplicative_impacts:
        yhat = yhat * impact
    for impact in additive_impacts:
        yhat = yhat + impact
    return yhat

def apply_inverse_impacts(
    y: tf.Tensor, multiplicative_impacts: list[tf.Tensor], additive_impacts: list[tf.Tensor]
) -> tf.Tensor:
    """Remove the impacts from `y` to get the baseline back.
    baseline = (yhat - Σ(additive effects)) / ∏(multiplicative effects)

    NOTE: Here its inverse impacts, i.e.,
        inverse multiplicative effect = 1/multiplicative effect
        inverse additive effect = -1 * additive effect

    Args:
        y (tf.Tensor): The `y` total impact.
        multiplicative_impacts (list[tf.Tensor]): Multiplicative impacts, are scaled down (larger effect).
        additive_impacts (list[tf.Tensor]): Additive impacts, are subtracted off the `y` (smaller effect).

    Returns:
        tf.Tensor: The baseline after all the impacts are removed from `y`.
    """
    baseline = y
    for impact in additive_impacts:
        baseline = baseline - impact
    for impact in multiplicative_impacts:
        baseline = tf.math.divide_no_nan(baseline, impact)
    return baseline