From c2cfa8d15e5d4e12a8ea0452e1069c1edca264bc Mon Sep 17 00:00:00 2001 From: Mickael GUIBERT <mickael.guibert@cea.fr> Date: Tue, 15 Apr 2025 12:37:32 +0000 Subject: [PATCH] [Fix](Producer) Add ignore attribute to False by default --- aidge_export_cpp/operators/Producer.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/aidge_export_cpp/operators/Producer.py b/aidge_export_cpp/operators/Producer.py index 1ae54d3..138b58c 100644 --- a/aidge_export_cpp/operators/Producer.py +++ b/aidge_export_cpp/operators/Producer.py @@ -47,7 +47,10 @@ class ProducerCPP(ExportNode): def __init__(self, node, mem_info): super().__init__(node, mem_info) self.values = np.array(self.operator.get_output(0)) - self.ignore = node.attributes().ignore + if node.attributes().has_attr("ignore"): + self.ignore = node.attributes().ignore + else: + self.ignore = False if len(self.values.shape) == 4: # Note: export in HWC self.values = np.transpose(self.values, (0, 2, 3, 1)) -- GitLab