From 785c5f7417d61f2d2f19ed9c781c8b90db24e161 Mon Sep 17 00:00:00 2001 From: Dmitry Gilemkhanov Date: Tue, 7 Jan 2025 16:41:23 +0300 Subject: [PATCH] Multi-modal test update --- test/unit/api/test_assumption_builder.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/test/unit/api/test_assumption_builder.py b/test/unit/api/test_assumption_builder.py index bb97b514d1..5830769e4d 100644 --- 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():