selftest - two sided

outbrain · Sep 23, 2023 · 59aef68 · 59aef68
1 parent 30960df
commit 59aef68
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/outrank/task_selftest.py b/outrank/task_selftest.py
@@ -31,7 +31,7 @@ def conduct_self_test():
     logger.info("Verifying output's properties ..")
     assert dfx.shape[0] == 201
     assert dfx.shape[1] == 3
-    assert dfx['FeatureA'].values.tolist().pop() == 'label-(81; 100)'
+    assert dfx['FeatureA'].values.tolist().pop() == 'label-(81; 100)' or dfx['FeatureB'].values.tolist().pop() == 'label-(81; 100)'
 
     to_remove = ['ranking_outputs', 'test_data_synthetic']
     for path in to_remove: