Skip to content

Commit e503ef0

Browse files
DRMPNLopa10ko
andcommitted
Apply suggestions from code review
Co-authored-by: George Lopatenko <81328772+Lopa10ko@users.noreply.github.com>
1 parent 79dbec8 commit e503ef0

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

test/integration/api/test_main_api.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -339,7 +339,7 @@ def test_forecast_with_not_ts_problem():
339339
def test_api_for_amlb(initial_assumption, timeout):
340340
amlb_data = Path(fedot_project_root(), 'test', 'data', 'amlb')
341341
x_train = np.load(str(Path(amlb_data, 'train_australian_fold7.npy')))
342-
y_train = np.load(str(Path(amlb_data, 'target_y.npy')), allow_pickle=True) # real target from ALMB
342+
y_train = np.load(str(Path(amlb_data, 'target_y.npy')), allow_pickle=True) # real target from AMLB
343343
x_test = np.load(str(Path(amlb_data, 'test_australian_fold7.npy')))
344344
# TODO resample add
345345
training_params = {"preset": "best_quality", "n_jobs": -1}

test/unit/api/test_main_api.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -138,7 +138,7 @@ def test_specific_baseline_with_api():
138138
assert len(prediction) == len(test_data)
139139

140140
baseline_metrics = baseline_model.get_metrics()
141-
assert baseline_metrics["f1"] > 0
141+
assert baseline_metrics["f1"]
142142

143143

144144
def test_forecast_with_multivariate_ts():

0 commit comments

Comments
 (0)