Skip to content

Commit 75bd079

Browse files
authored
Bug Fix: Update uplift.pyx with isinstance(v, Numbers.number)) (#849)
* Update uplift.pyx with isinstance(v, numbers.Number):
1 parent de4dcfc commit 75bd079

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

causalml/inference/tree/uplift.pyx

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@ from joblib import Parallel, delayed
3333
from packaging import version
3434
from sklearn.model_selection import train_test_split
3535
from sklearn.utils import check_X_y, check_array, check_random_state
36+
import numbers
3637

3738
if version.parse(sklearn.__version__) >= version.parse('0.22.0'):
3839
from sklearn.utils._testing import ignore_warnings
@@ -2264,7 +2265,7 @@ class UpliftTreeClassifier:
22642265
else:
22652266
v = observations[tree.col]
22662267
branch = None
2267-
if isinstance(v, int) or isinstance(v, float):
2268+
if isinstance(v, numbers.Number):
22682269
if v >= tree.value:
22692270
branch = tree.trueBranch
22702271
else:
@@ -2311,7 +2312,7 @@ class UpliftTreeClassifier:
23112312
return dict(result)
23122313
else:
23132314
branch = None
2314-
if isinstance(v, int) or isinstance(v, float):
2315+
if isinstance(v, numbers.Number):
23152316
if v >= tree.value:
23162317
branch = tree.trueBranch
23172318
else:

0 commit comments

Comments
 (0)