Skip to content
Snippets Groups Projects
Commit ff1ac55a authored by Axel Farrugia's avatar Axel Farrugia
Browse files

[Fix](Exports) Add `test_mode` as argument to the template while generating the forward file

parent fc986049
No related branches found
No related tags found
2 merge requests!279v0.4.0,!250[Feat](Exports) Add custom options to exports
......@@ -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):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment