Skip to content

Commit f7dcc98

Browse files
committed
update type hint for cv parameter in PSProcessor to use Union[int, list]
1 parent c69af49 commit f7dcc98

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

doubleml/utils/propensity_score_processing.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -171,7 +171,7 @@ def _apply_calibration(
171171
self,
172172
propensity_scores: np.ndarray,
173173
treatment: np.ndarray,
174-
cv: Optional[int | list] = None,
174+
cv: Optional[Union[int, list]] = None,
175175
) -> np.ndarray:
176176
"""Apply calibration method to propensity scores if specified."""
177177
if self.calibration_method is None:

0 commit comments

Comments
 (0)