Skip to content

Commit b9876f5

Browse files
Update docstring
1 parent e14a7f8 commit b9876f5

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

balance/utils/model_matrix.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -308,9 +308,10 @@ def _prepare_input_model_matrix(
308308
unique. Defaults to True.
309309
310310
Raises:
311-
ValueError: If requested ``variables`` are not present in both inputs,
312-
if variables contain ``[`` or ``]``, or if ``add_na=False`` drops
313-
all rows from sample/target.
311+
ValueError: If requested ``variables`` are not present in the
312+
provided input frame(s) (and in both sample and target when target
313+
is supplied), if variables contain ``[`` or ``]``, or if
314+
``add_na=False`` drops all rows from sample/target.
314315
315316
Returns:
316317
Dict[str, Any]: returns a dictionary containing two keys: 'all_data' and 'sample_n'.

0 commit comments

Comments
 (0)