Apply suggestions from code review

Co-authored-by: George Lopatenko <81328772+Lopa10ko@users.noreply.github.com>
aimclub · Apr 11, 2024 · e503ef0 · e503ef0
1 parent 79dbec8
commit e503ef0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/test/integration/api/test_main_api.py b/test/integration/api/test_main_api.py
@@ -339,7 +339,7 @@ def test_forecast_with_not_ts_problem():
 def test_api_for_amlb(initial_assumption, timeout):
     amlb_data = Path(fedot_project_root(), 'test', 'data', 'amlb')
     x_train = np.load(str(Path(amlb_data, 'train_australian_fold7.npy')))
-    y_train = np.load(str(Path(amlb_data, 'target_y.npy')), allow_pickle=True)  # real target from ALMB
+    y_train = np.load(str(Path(amlb_data, 'target_y.npy')), allow_pickle=True)  # real target from AMLB
     x_test = np.load(str(Path(amlb_data, 'test_australian_fold7.npy')))
     # TODO resample add
     training_params = {"preset": "best_quality", "n_jobs": -1}

diff --git a/test/unit/api/test_main_api.py b/test/unit/api/test_main_api.py
@@ -138,7 +138,7 @@ def test_specific_baseline_with_api():
     assert len(prediction) == len(test_data)
 
     baseline_metrics = baseline_model.get_metrics()
-    assert baseline_metrics["f1"] > 0
+    assert baseline_metrics["f1"]
 
 
 def test_forecast_with_multivariate_ts():