From daa641e18d49a7b15d7bb12a613e6c394dd47a61 Mon Sep 17 00:00:00 2001
From: cmoineau <cyril.moineau@cea.fr>
Date: Thu, 26 Oct 2023 08:42:11 +0000
Subject: [PATCH] [NodeExport] Update outputs_dims.
---
aidge_core/export/node_export.py | 6 +-----
1 file changed, 1 insertion(+), 5 deletions(-)
diff --git a/aidge_core/export/node_export.py b/aidge_core/export/node_export.py
index 794fec814..32629ae51 100644
--- a/aidge_core/export/node_export.py
+++ b/aidge_core/export/node_export.py
@@ -41,13 +41,9 @@ class ExportNode(ABC):
else:
self.inputs_dims.append(None)
- # TODO
for idx, child_node in enumerate(self.node.get_children()):
self.outputs.append(child_node)
- if child_node is not None:
- self.outputs_dims.append(self.operator.output(idx).dims())
- else:
- self.outputs_dims.append(None)
+ self.outputs_dims.append(self.operator.output(idx).dims())
@abstractmethod
def export(self, export_folder:str, list_configs:list):
--
GitLab