From 573425cf5fb8eee067a62beaf3b5ec734afe989a Mon Sep 17 00:00:00 2001
From: Axel Farrugia <axel.farrugia@cea.fr>
Date: Fri, 2 May 2025 13:49:42 +0200
Subject: [PATCH] [Refactor](R18) Adapt to new aidge_cmp flag system

---
 examples/export_ResNet18/resnet18.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/examples/export_ResNet18/resnet18.py b/examples/export_ResNet18/resnet18.py
index f60e68a..64b3a9b 100644
--- a/examples/export_ResNet18/resnet18.py
+++ b/examples/export_ResNet18/resnet18.py
@@ -502,8 +502,9 @@ This option has to be passed to each node in order to be used within the Export
 (JConv, JPad, ...) that you can find in the "export_gen/operator_export" folder. 
 """
 
-for node in model.get_nodes():
-    node.attributes().aidge_cmp = AIDGE_CMP
+if AIDGE_CMP:
+    for node in model.get_nodes():
+        node.attributes().aidge_cmp = True
 
 # --------------------------------------------------------------
 # EXPORT THE MODEL
-- 
GitLab