From eb0e9ed9daad474cd579b634479005b89416a27e Mon Sep 17 00:00:00 2001
From: NAUD Maxence <maxence.naud@cea.fr>
Date: Fri, 5 Apr 2024 10:10:21 +0000
Subject: [PATCH] fix node_export.py for new scheduling that does not contain
input Producer
---
aidge_core/export/node_export.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/aidge_core/export/node_export.py b/aidge_core/export/node_export.py
index bea61551d..477989b03 100644
--- a/aidge_core/export/node_export.py
+++ b/aidge_core/export/node_export.py
@@ -39,7 +39,11 @@ class ExportNode(ABC):
if parent_node is not None:
self.inputs_dims.append(self.operator.get_input(idx).dims())
else:
- self.inputs_dims.append(None)
+ print(self.operator.get_input(idx))
+ if self.operator.get_input(idx) is not None:
+ self.inputs_dims.append(self.operator.get_input(idx).dims())
+ else:
+ self.inputs_dims.append(None)
for idx, child_node in enumerate(self.node.get_children()):
self.outputs.append(child_node)
--
GitLab