Skip to content

Commit

Permalink
remove unused argument to marginal_pit
Browse files Browse the repository at this point in the history
  • Loading branch information
elray1 committed Oct 22, 2024
1 parent 0c7eedd commit 336ae90
Showing 1 changed file with 2 additions and 5 deletions.
7 changes: 2 additions & 5 deletions src/postpredict/metrics.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,8 +77,8 @@ def energy_score_one_unit(df: pl.DataFrame):


def marginal_pit(model_out_wide: pl.DataFrame, obs_data_wide: pl.DataFrame,
key_cols: list[str] | None, pred_cols: list[str], obs_cols: list[str],
reduce_mean: bool = True) -> float | pl.DataFrame:
key_cols: list[str] | None, pred_cols: list[str],
obs_cols: list[str]) -> pl.DataFrame:
"""
Compute the probability integral transform (PIT) value for each of a
collection of marginal predictive distributions represented by a set of
Expand All @@ -102,9 +102,6 @@ def marginal_pit(model_out_wide: pl.DataFrame, obs_data_wide: pl.DataFrame,
obs_cols: list[str]
Columns that appear in `obs_data_wide` and identify observed values. The
order of these should match the order of `pred_cols`.
reduce_mean: bool = True
Indicator of whether to return a numeric mean energy score (default) or
a pl.DataFrame with one row per observational unit.
Returns
-------
Expand Down

0 comments on commit 336ae90

Please sign in to comment.