[catboost] Fix test.
@@ -140,7 +140,7 @@ def _predict_python_on_test(dataset, apply_catboost_model):
pred_python = []
for _, row in features_data.iterrows():
- float_features = [float(v) for v in row.values[float_feature_indices]]
+ float_features = [round(v, 9) for v in row.values[float_feature_indices]]
cat_features = row.values[cat_feature_indices]
pred_python.append(apply_catboost_model(float_features, cat_features))
return pred_python
@@ -164,7 +164,6 @@ def test_python_export_from_app(dataset):
@pytest.mark.parametrize('iterations', [2, 40])
@pytest.mark.parametrize('dataset', ['adult', 'higgs'])
-@pytest.mark.xfail(reason='MLTOOLS-4300')
def test_python_export_from_python(dataset, iterations):
train_pool, test_pool = _get_train_test_pool(dataset)
@@ -183,7 +182,6 @@ def test_python_export_from_python(dataset, iterations):
@pytest.mark.parametrize('dataset', ['adult', 'higgs'])
-@pytest.mark.xfail(reason='MLTOOLS-4300')
def test_python_after_load(dataset):
train_pool, test_pool = _get_train_test_pool(dataset)
model = CatBoostClassifier(iterations=40, random_seed=0)
