fix pipeline_preprocessing_through_api_correctly

aimclub · Oct 6, 2023 · 2791288 · 2791288
1 parent 153ce40
commit 2791288
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/integration/api/test_main_api.py b/test/integration/api/test_main_api.py
@@ -501,7 +501,8 @@ def test_pipeline_preprocessing_through_api_correctly():
     # Stand-alone pipeline with it's own preprocessing
     predicted = pipeline.predict(data, output_mode='labels')
 
-    assert predicted.predict[-1] == 'green-orange'
+    # check whether NaN-field was correctly predicted
+    assert predicted.predict[3] == 'red-blue'
 
 
 def test_data_from_csv_load_correctly():