Skip to content

Commit c69af49

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

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

doubleml/utils/propensity_score_processing.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import warnings
22
from dataclasses import dataclass
3-
from typing import Optional
3+
from typing import Optional, Union
44

55
import numpy as np
66
from sklearn.isotonic import IsotonicRegression
@@ -131,7 +131,7 @@ def adjust_ps(
131131
self,
132132
propensity_scores: np.ndarray,
133133
treatment: np.ndarray,
134-
cv: Optional[int | list] = None,
134+
cv: Optional[Union[int, list]] = None,
135135
learner_name: Optional[str] = None,
136136
) -> np.ndarray:
137137
"""

0 commit comments

Comments
 (0)