diff --git a/aidge_export_cpp/export_utils.py b/aidge_export_cpp/export_utils.py index a7f32bb514b963e6795638b4e0f9ba9686e1eec7..9ce15f5ec9ca3797a1a885c648b1c64160fe0010 100644 --- a/aidge_export_cpp/export_utils.py +++ b/aidge_export_cpp/export_utils.py @@ -167,7 +167,6 @@ def exclude_unwanted_producers(model): nodes_to_ignore = ["Mul", "BitShift"] for node in model.get_nodes(): - node.attributes().ignore = False if node.type() == "Producer": children_nodes = [n.type() for n in node.get_children()] for node_type in nodes_to_ignore: diff --git a/aidge_export_cpp/operators/Producer.py b/aidge_export_cpp/operators/Producer.py index 6c7547101e40a5a08c9c08ca4319f8d0966cd1b3..627dcb29ed21e2779a30978b5b22768c0c2cc9c4 100644 --- a/aidge_export_cpp/operators/Producer.py +++ b/aidge_export_cpp/operators/Producer.py @@ -47,15 +47,7 @@ class ProducerCPP(ExportNode): def __init__(self, node, mem_info): super().__init__(node, mem_info) self.values = np.array(self.operator.get_output(0)) - if node.attributes().has_attr("ignore"): - self.ignore = node.attributes().ignore - else: - self.ignore = False - - if node.attributes().has_attr("ignore"): - self.ignore = node.attributes().ignore - else: - self.ignore = False + self.ignore = node.attributes().has_attr("ignore") if len(self.values.shape) == 4: # Note: export in HWC self.values = np.transpose(self.values, (0, 2, 3, 1))