Skip to content

Commit 00464ae

Browse files
Remove GPU restriction from test_xgboost.py
1 parent 9813fda commit 00464ae

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

tests/unit/systems/dag/runtimes/local/ops/fil/test_xgboost.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22
import pandas as pd
33
import pytest
44

5-
from merlin.core.compat import HAS_GPU
65
from merlin.dag import ColumnSelector
76
from merlin.dtypes.shape import Shape
87
from merlin.schema import ColumnSchema, Schema
@@ -17,7 +16,6 @@
1716

1817

1918
@pytest.mark.parametrize("runtime", [Runtime()])
20-
@pytest.mark.skipif(not HAS_GPU, reason="no gpu detected")
2119
def test_xgboost_regressor_forest_inference(runtime, tmpdir):
2220
rows = 200
2321
num_features = 16
@@ -52,7 +50,6 @@ def test_xgboost_regressor_forest_inference(runtime, tmpdir):
5250
assert response["output__0"].shape == Shape((5,))
5351

5452

55-
@pytest.mark.skipif(not HAS_GPU, reason="no gpu detected")
5653
@pytest.mark.parametrize("runtime", [Runtime()])
5754
def test_xgboost_classify_forest_inference(runtime, tmpdir):
5855
rows = 200

0 commit comments

Comments
 (0)