From ff1ac55a7d06c54b7abec212596e480454452164 Mon Sep 17 00:00:00 2001
From: Axel Farrugia <axel.farrugia@cea.fr>
Date: Thu, 14 Nov 2024 12:06:17 +0100
Subject: [PATCH] [Fix](Exports) Add `test_mode` as argument to the template
while generating the forward file
---
aidge_core/export_utils/scheduler_export.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/aidge_core/export_utils/scheduler_export.py b/aidge_core/export_utils/scheduler_export.py
index 310e7e650..98c78b087 100644
--- a/aidge_core/export_utils/scheduler_export.py
+++ b/aidge_core/export_utils/scheduler_export.py
@@ -172,7 +172,8 @@ def scheduler_export(scheduler, export_folder_path: str, export_lib: ExportLib =
inputs_name=inputs_name,
inputs_dtype=inputs_dtype,
outputs_name=outputs_name,
- outputs_dtype=outputs_dtype
+ outputs_dtype=outputs_dtype,
+ test_mode=test_mode
)
forward_header_template = str(ROOT / "templates" / "forward_header.jinja")
@@ -188,7 +189,8 @@ def scheduler_export(scheduler, export_folder_path: str, export_lib: ExportLib =
inputs_name=inputs_name,
inputs_dtype=inputs_dtype,
outputs_name=outputs_name,
- outputs_dtype=outputs_dtype
+ outputs_dtype=outputs_dtype,
+ test_mode=test_mode
)
if len(outputs_name) != len(outputs_dtype) or len(outputs_name) != len(outputs_size):
--
GitLab