Skip to content

Commit bcdd3fc

Browse files
committed
Update xfail list
1 parent 04a6138 commit bcdd3fc

3 files changed

Lines changed: 65 additions & 82 deletions

File tree

python/cuml/cuml/accel/_wrappers/sklearn/ensemble.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,15 @@ def _gpu_score(self, X, y, sample_weight=None):
3434
raise UnsupportedOnGPU
3535
return self._gpu.score(X, y)
3636

37+
def __len__(self):
38+
return self._call_method("__len__")
39+
40+
def __iter__(self):
41+
return self._call_method("__iter__")
42+
43+
def __getitem__(self, index):
44+
return self._call_method("__getitem__", index)
45+
3746

3847
class RandomForestClassifier(ProxyBase):
3948
_gpu_class = cuml.ensemble.RandomForestClassifier
@@ -47,3 +56,12 @@ def _gpu_score(self, X, y, sample_weight=None):
4756
if sample_weight is not None:
4857
raise UnsupportedOnGPU
4958
return self._gpu.score(X, y)
59+
60+
def __len__(self):
61+
return self._call_method("__len__")
62+
63+
def __iter__(self):
64+
return self._call_method("__iter__")
65+
66+
def __getitem__(self, index):
67+
return self._call_method("__getitem__", index)

python/cuml/cuml/accel/pytest_plugin.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -106,6 +106,7 @@ def pytest_collection_modifyitems(config, items):
106106

107107
reason = group["reason"]
108108
strict = group.get("strict", True)
109+
run = group.get("run", True)
109110
tests = group["tests"]
110111
condition = True
111112
if "condition" in group:
@@ -115,6 +116,7 @@ def pytest_collection_modifyitems(config, items):
115116
config = {
116117
"reason": reason,
117118
"strict": strict,
119+
"run": run,
118120
"condition": condition,
119121
"extra_marker": marker,
120122
}
@@ -131,6 +133,7 @@ def pytest_collection_modifyitems(config, items):
131133
pytest.mark.xfail(
132134
reason=config["reason"],
133135
strict=config["strict"],
136+
run=config["run"],
134137
condition=config["condition"],
135138
)
136139
)

0 commit comments

Comments
 (0)