fix tests

Author: reuvenp

tests/keras_tests/function_tests/test_activation_weights_composition_substitution.py

@@ -227,10 +227,17 @@ def test_all_weights_layers_composition(self):
                               base_config=base_config,
                               default_config=default_config)
 
+        # Validation is skipped because fusing information is not relevant for the virtual graph.
+        # Therefore, validation checks are disabled before the virtual graph substitution and
+        # re-enabled once it completes.
+        graph.skip_validation_check = True
+
         # Nodes split and composition substitution
         split_graph = substitute(copy.deepcopy(graph), [WeightsActivationSplit()])
         v_graph = substitute(copy.deepcopy(split_graph), [VirtualActivationWeightsComposition()])
 
+        graph.skip_validation_check = False
+
         assert split_graph is not graph
         self.assertTrue(len(v_graph.nodes) == 8)
         self.assertTrue(len([n for n in v_graph.nodes if isinstance(n, VirtualActivationWeightsNode)]) == 5)

tests/keras_tests/function_tests/test_cfg_candidates_filter.py

@@ -58,8 +58,10 @@ def prepare_graph(in_model, base_config, default_config, bitwidth_candidates):
     graph = set_quantization_configuration_to_graph(graph=graph,
                                                     quant_config=mct.core.QuantizationConfig(),
                                                     mixed_precision_enable=True)
+
     fusing_info = FusingInfoGenerator(fqc.get_fusing_patterns()).generate_fusing_info(graph)
-    graph = FusingMetadataWrapper(graph, fusing_info)
+    graph.fusing_info = fusing_info
+    graph.disable_fused_nodes_activation_quantization()
 
     return graph