Skip to content

Commit

Permalink
Daily run failure: add n_samples keyword argument to `ColumnTransfo…
Browse files Browse the repository at this point in the history
…rmer` in benchmarks (#767)

* add n_samples keyword argument

* not required

* fix

* type check

* finally

* remove auxiliary assignment to _
  • Loading branch information
MatthiasSchmidtblaicherQC authored Feb 20, 2024
1 parent 2e39441 commit 3ba4209
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/glum_benchmarks/data/create_insurance.py
Original file line number Diff line number Diff line change
Expand Up @@ -180,7 +180,7 @@ def __init__(
transformer_weights=transformer_weights,
)

def _hstack(self, Xs: Iterable[Union[pd.Series, pd.DataFrame]]):
def _hstack(self, Xs: Iterable[Union[pd.Series, pd.DataFrame]], *, n_samples=None):
"""Stacks X horizontally."""
return pd.concat(Xs, axis="columns")

Expand Down

0 comments on commit 3ba4209

Please sign in to comment.