Skip to content

Commit d38cb53

Browse files
committed
Add skips for lightgbm and treelite in tests
1 parent 30e8b29 commit d38cb53

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

improver_tests/calibration/rainforests_training/test_CompileRainForestsCalibration.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,10 @@
1111
CompileRainForestsCalibration,
1212
)
1313

14+
lightgbm = pytest.importorskip("lightgbm")
15+
tl2cgen = pytest.importorskip("tl2cgen")
16+
treelite = pytest.importorskip("treelite")
17+
1418

1519
def test__init__(lightgbm_available, treelite_available, tmp_path):
1620
"""Test class is created if treelight libraries are available.

improver_tests/calibration/rainforests_training/test_TrainRainForestsCalibration.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,8 @@
1111
TrainRainForestsCalibration,
1212
)
1313

14+
lightgbm = pytest.importorskip("lightgbm")
15+
1416

1517
def test__init__lightgmb_available(lightgbm_available):
1618
"""Test class is created if lightgbm library is available.

0 commit comments

Comments
 (0)