From 49c11d9da77e847227195b192e70f27d3114f024 Mon Sep 17 00:00:00 2001 From: TomeHirata Date: Wed, 19 Aug 2026 16:07:12 +0900 Subject: [PATCH] Fix predict() docstring to label argument as locations The `predict()` method in `DistributionEstimatorBase` takes a `locations` argument but the docstring incorrectly labelled it as `outcomes`. Co-Authored-By: Claude Opus 4.7 (1M context) --- dte_adj/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dte_adj/base.py b/dte_adj/base.py index 4650cc0..cfa6605 100644 --- a/dte_adj/base.py +++ b/dte_adj/base.py @@ -424,7 +424,7 @@ def predict( Args: treatment_arm (int): The index of the treatment arm. - outcomes (np.ndarray): Scalar values to be used for computing the cumulative distribution. + locations (np.ndarray): Scalar values to be used for computing the cumulative distribution. display_progress (bool, optional): Whether to display a progress bar. Defaults to True. Returns: