From 943410649b34ec42c3b22195dd80a2ab2adffb9d Mon Sep 17 00:00:00 2001 From: Sergei Pakulin Date: Mon, 9 Oct 2023 17:17:39 +0500 Subject: [PATCH] change api_predict_correct train_data --- test/integration/api/test_main_api.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/integration/api/test_main_api.py b/test/integration/api/test_main_api.py index 319ab82e49..fe743970d5 100644 --- a/test/integration/api/test_main_api.py +++ b/test/integration/api/test_main_api.py @@ -153,7 +153,7 @@ def data_with_binary_features_and_categorical_target(): @pytest.mark.parametrize('task_type, metric_name', [ ('classification', 'f1'), - ('regression', 'rmse'), + ('regression', 'rmse') ]) def test_api_predict_correct(task_type, metric_name): train_data, test_data, _ = get_dataset(task_type) @@ -167,7 +167,7 @@ def test_api_predict_correct(task_type, metric_name): # composing and tuning was applied assert model.history is not None assert model.history.tuning_result is not None - assert is_predict_ignores_target(model.predict, train_data, 'features') + assert is_predict_ignores_target(model.predict, model.train_data, 'features') @pytest.mark.parametrize('task_type, metric_name, pred_model', [