Multi-modal test update

aimclub · Jan 7, 2025 · 785c5f7 · 785c5f7
1 parent 8fcc017
commit 785c5f7
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/test/unit/api/test_assumption_builder.py b/test/unit/api/test_assumption_builder.py
@@ -95,8 +95,7 @@ def test_assumptions_builder_for_multimodal_data():
     assert pipeline_contains_all(mm_pipeline, *mm_data)
     assert len(list(filter(lambda node: node.is_primary, mm_pipeline.nodes)))
     assert len(mm_pipeline.root_node.nodes_from) == mm_data.num_classes
-    # TODO: do we need this assert?
-    # assert mm_pipeline.length == mm_pipeline.depth * len(mm_data) - 1  # minus final ensemble
+    assert mm_pipeline.length > len(mm_data)
 
 
 def test_assumptions_builder_unsuitable_available_operations():