@@ -208,7 +208,10 @@ def test_forest_regression(cas_session, boston_dataset):
208
208
props = _get_model_properties (desc )
209
209
210
210
for k , v in target .items ():
211
- assert props [k ] == v
211
+ if k == "algorithm" :
212
+ assert props [k ] in ("Random forest" , "Tree-based model" )
213
+ else :
214
+ assert props [k ] == v
212
215
213
216
files = create_files_from_astore (cas_session .CASTable ("astore" ))
214
217
check_input_variables (files , BOSTON_INPUT_VARS )
@@ -239,7 +242,10 @@ def test_forest_regression_with_nominals(cas_session, boston_dataset):
239
242
props = _get_model_properties (desc )
240
243
241
244
for k , v in target .items ():
242
- assert props [k ] == v
245
+ if k == "algorithm" :
246
+ assert props [k ] in ("Random forest" , "Tree-based model" )
247
+ else :
248
+ assert props [k ] == v
243
249
244
250
files = create_files_from_astore (cas_session .CASTable ("astore" ))
245
251
check_input_variables (files , BOSTON_INPUT_VARS )
@@ -268,7 +274,10 @@ def test_gradboost_binary_classification(cas_session, cancer_dataset):
268
274
props = _get_model_properties (desc )
269
275
270
276
for k , v in target .items ():
271
- assert props [k ] == v
277
+ if k == "algorithm" :
278
+ assert props [k ] in ("Gradient boosting" , "Tree-based model" )
279
+ else :
280
+ assert props [k ] == v
272
281
273
282
files = create_files_from_astore (cas_session .CASTable ("astore" ))
274
283
check_input_variables (files , CANCER_INPUT_VARS )
@@ -296,7 +305,10 @@ def test_gradboost_classification(cas_session, iris_dataset):
296
305
props = _get_model_properties (desc )
297
306
298
307
for k , v in target .items ():
299
- assert props [k ] == v
308
+ if k == "algorithm" :
309
+ assert props [k ] in ("Gradient boosting" , "Tree-based model" )
310
+ else :
311
+ assert props [k ] == v
300
312
301
313
files = create_files_from_astore (cas_session .CASTable ("astore" ))
302
314
check_input_variables (files , IRIS_INPUT_VARS )
@@ -324,7 +336,10 @@ def test_gradboost_regression(cas_session, boston_dataset):
324
336
props = _get_model_properties (desc )
325
337
326
338
for k , v in target .items ():
327
- assert props [k ] == v
339
+ if k == "algorithm" :
340
+ assert props [k ] in ("Gradient boosting" , "Tree-based model" )
341
+ else :
342
+ assert props [k ] == v
328
343
329
344
files = create_files_from_astore (cas_session .CASTable ("astore" ))
330
345
check_input_variables (files , BOSTON_INPUT_VARS )
@@ -355,7 +370,10 @@ def test_gradboost_regression_with_nominals(cas_session, boston_dataset):
355
370
props = _get_model_properties (desc )
356
371
357
372
for k , v in target .items ():
358
- assert props [k ] == v
373
+ if k == "algorithm" :
374
+ assert props [k ] in ("Gradient boosting" , "Tree-based model" )
375
+ else :
376
+ assert props [k ] == v
359
377
360
378
files = create_files_from_astore (cas_session .CASTable ("astore" ))
361
379
check_input_variables (files , BOSTON_INPUT_VARS )
0 commit comments