File tree Expand file tree Collapse file tree 4 files changed +3
-3
lines changed Expand file tree Collapse file tree 4 files changed +3
-3
lines changed File renamed without changes.
File renamed without changes.
File renamed without changes.
Original file line number Diff line number Diff line change @@ -21,7 +21,7 @@ def test_xgb2ds():
21
21
pytest .importorskip ('xgboost' )
22
22
23
23
IN_PKL = os .path .join (DATA_PATH , 'xgb.pkl' )
24
- EXPECTED_SAS = os .path .join (DATA_PATH , 'xgb.sas ' )
24
+ EXPECTED_SAS = os .path .join (DATA_PATH , 'xgb_datastep ' )
25
25
26
26
from sasctl .utils .pyml2ds .connectors .ensembles .xgb import XgbTreeParser
27
27
@@ -50,7 +50,7 @@ def test_lgb2ds():
50
50
pytest .importorskip ('lightgbm' )
51
51
52
52
IN_PKL = os .path .join (DATA_PATH , 'lgb.pkl' )
53
- EXPECTED_SAS = os .path .join (DATA_PATH , 'lgb.sas ' )
53
+ EXPECTED_SAS = os .path .join (DATA_PATH , 'lgb_datastep ' )
54
54
55
55
from sasctl .utils .pyml2ds .connectors .ensembles .lgb import LightgbmTreeParser
56
56
@@ -77,7 +77,7 @@ def _leaf_value(self):
77
77
78
78
def test_gbm2ds ():
79
79
IN_PKL = os .path .join (DATA_PATH , 'gbm.pmml' )
80
- EXPECTED_SAS = os .path .join (DATA_PATH , 'gbm.sas ' )
80
+ EXPECTED_SAS = os .path .join (DATA_PATH , 'gbm_datastep ' )
81
81
82
82
from sasctl .utils .pyml2ds .connectors .ensembles .pmml import PmmlTreeParser
83
83
You can’t perform that action at this time.
0 commit comments