@@ -133,6 +133,7 @@ def _build_and_save_xgboost(
133
133
)
134
134
@pytest .mark .parametrize ("n_classes" , [2 , 5 , 25 ])
135
135
@pytest .mark .skipif (has_xgboost () is False , reason = "need to install xgboost" )
136
+ @pytest .mark .xfail (reason = "xgboost's allocator doesn't work with older drivers" )
136
137
@pytest .mark .filterwarnings ("ignore::FutureWarning" )
137
138
def test_fil_classification (
138
139
n_rows , n_columns , num_rounds , n_classes , tmp_path
@@ -205,6 +206,7 @@ def test_fil_classification(
205
206
"max_depth" , [unit_param (3 ), unit_param (7 ), stress_param (11 )]
206
207
)
207
208
@pytest .mark .skipif (has_xgboost () is False , reason = "need to install xgboost" )
209
+ @pytest .mark .xfail (reason = "xgboost's allocator doesn't work with older drivers" )
208
210
@pytest .mark .filterwarnings ("ignore::FutureWarning" )
209
211
def test_fil_regression (n_rows , n_columns , num_rounds , tmp_path , max_depth ):
210
212
# settings
@@ -469,6 +471,7 @@ def small_classifier_and_preds(tmpdir_factory, request):
469
471
470
472
471
473
@pytest .mark .skipif (has_xgboost () is False , reason = "need to install xgboost" )
474
+ @pytest .mark .xfail (reason = "xgboost's allocator doesn't work with older drivers" )
472
475
@pytest .mark .parametrize (
473
476
"algo" ,
474
477
[
@@ -501,6 +504,7 @@ def test_output_algos(algo, small_classifier_and_preds):
501
504
502
505
503
506
@pytest .mark .skipif (has_xgboost () is False , reason = "need to install xgboost" )
507
+ @pytest .mark .xfail (reason = "xgboost's allocator doesn't work with older drivers" )
504
508
@pytest .mark .parametrize ("precision" , ["native" , "float32" , "float64" ])
505
509
@pytest .mark .filterwarnings ("ignore::FutureWarning" )
506
510
def test_precision_xgboost (precision , small_classifier_and_preds ):
@@ -521,6 +525,7 @@ def test_precision_xgboost(precision, small_classifier_and_preds):
521
525
522
526
523
527
@pytest .mark .skipif (has_xgboost () is False , reason = "need to install xgboost" )
528
+ @pytest .mark .xfail (reason = "xgboost's allocator doesn't work with older drivers" )
524
529
@pytest .mark .parametrize (
525
530
"storage_type" , [False , True , "auto" , "dense" , "sparse" , "sparse8" ]
526
531
)
@@ -543,6 +548,7 @@ def test_output_storage_type(storage_type, small_classifier_and_preds):
543
548
544
549
545
550
@pytest .mark .skipif (has_xgboost () is False , reason = "need to install xgboost" )
551
+ @pytest .mark .xfail (reason = "xgboost's allocator doesn't work with older drivers" )
546
552
@pytest .mark .parametrize ("storage_type" , ["dense" , "sparse" ])
547
553
@pytest .mark .parametrize ("blocks_per_sm" , [1 , 2 , 3 , 4 ])
548
554
@pytest .mark .filterwarnings ("ignore::FutureWarning" )
@@ -567,6 +573,7 @@ def test_output_blocks_per_sm(
567
573
568
574
569
575
@pytest .mark .skipif (has_xgboost () is False , reason = "need to install xgboost" )
576
+ @pytest .mark .xfail (reason = "xgboost's allocator doesn't work with older drivers" )
570
577
@pytest .mark .parametrize ("threads_per_tree" , [2 , 4 , 8 , 16 , 32 , 64 , 128 , 256 ])
571
578
@pytest .mark .filterwarnings ("ignore::FutureWarning" )
572
579
def test_threads_per_tree (threads_per_tree , small_classifier_and_preds ):
@@ -594,6 +601,7 @@ def test_threads_per_tree(threads_per_tree, small_classifier_and_preds):
594
601
595
602
@pytest .mark .parametrize ("output_class" , [True , False ])
596
603
@pytest .mark .skipif (has_xgboost () is False , reason = "need to install xgboost" )
604
+ @pytest .mark .xfail (reason = "xgboost's allocator doesn't work with older drivers" )
597
605
@pytest .mark .filterwarnings ("ignore::FutureWarning" )
598
606
def test_thresholding (output_class , small_classifier_and_preds ):
599
607
model_path , model_type , X , xgb_preds = small_classifier_and_preds
@@ -612,6 +620,7 @@ def test_thresholding(output_class, small_classifier_and_preds):
612
620
613
621
614
622
@pytest .mark .skipif (has_xgboost () is False , reason = "need to install xgboost" )
623
+ @pytest .mark .xfail (reason = "xgboost's allocator doesn't work with older drivers" )
615
624
@pytest .mark .filterwarnings ("ignore::FutureWarning" )
616
625
def test_output_args (small_classifier_and_preds ):
617
626
model_path , model_type , X , xgb_preds = small_classifier_and_preds
0 commit comments