File tree Expand file tree Collapse file tree 1 file changed +0
-3
lines changed
tests/unit/systems/dag/runtimes/local/ops/fil Expand file tree Collapse file tree 1 file changed +0
-3
lines changed Original file line number Diff line number Diff line change 22import pandas as pd
33import pytest
44
5- from merlin .core .compat import HAS_GPU
65from merlin .dag import ColumnSelector
76from merlin .dtypes .shape import Shape
87from merlin .schema import ColumnSchema , Schema
1716
1817
1918@pytest .mark .parametrize ("runtime" , [Runtime ()])
20- @pytest .mark .skipif (not HAS_GPU , reason = "no gpu detected" )
2119def 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 ()])
5754def test_xgboost_classify_forest_inference (runtime , tmpdir ):
5855 rows = 200
You can’t perform that action at this time.
0 commit comments