diff --git a/aidge_export_cpp/export.py b/aidge_export_cpp/export.py index de20f3b4eee728a48cdef7849cd5e4894ee0e3d1..b8fe28a54fd5ccc2cd8e380722b431b51a3e1aa8 100644 --- a/aidge_export_cpp/export.py +++ b/aidge_export_cpp/export.py @@ -77,7 +77,7 @@ def export(export_folder_name, graphview, scheduler): list_inputs_name = [] for node in graphview.get_nodes(): if node.type() == "Producer": - if not node.get_operator().get_attr("Constant"): + if not node.get_operator().attr.constant: export_type = aidge_datatype2ctype(node.get_operator().get_output(0).dtype()) list_inputs_name.append((export_type, node.name())) diff --git a/examples/add_custom_operator/add_custom_operator.ipynb b/examples/add_custom_operator/add_custom_operator.ipynb index 0687e322831f045a17e729b2706ef518f3db6a7a..5477cdaefa00334a64472810e71c04031c06ab0f 100644 --- a/examples/add_custom_operator/add_custom_operator.ipynb +++ b/examples/add_custom_operator/add_custom_operator.ipynb @@ -89,7 +89,7 @@ "model.add(input_node)\n", "\n", "# Configuration for the model + forward dimensions\n", - "model.compile(\"cpu\", aidge_core.DataType.Float32)" + "model.compile(\"cpu\", aidge_core.dtype.float32)" ] }, { @@ -231,9 +231,9 @@ " def __init__(self, node):\n", " super().__init__(node)\n", "\n", - " self.betas = self.operator.get_attr(\"betas\")\n", + " self.betas: float = self.operator.get_attr(\"betas\")\n", "\n", - " def export(self, export_folder:str, list_configs:list):\n", + " def export(self, export_folder: str, list_configs: list[str]) -> list[str]:\n", "\n", " copyfile(\"for_export/swish_kernel.hpp\",\n", " f\"{export_folder}/include/kernels/\")\n", @@ -248,7 +248,7 @@ "\n", " return list_configs\n", "\n", - " def forward(self, list_actions:list):\n", + " def forward(self, list_actions:list[str]) -> list[str]:\n", "\n", " if not self.is_last:\n", " list_actions.append(set_up_output(self.name, \"float\"))\n", @@ -296,7 +296,7 @@ "outputs": [], "source": [ "digit = np.load(\"digit.npy\")\n", - "cpp.generate_input_file(\"inputs\", digit.reshape(-1), \"myexport/inputs.h\")" + "cpp.generate_input_file(array_name=\"inputs\", array=digit.reshape(-1), folder_path=\"myexport\")" ] }, { @@ -334,7 +334,7 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.9.16" + "version": "3.1.undefined" } }, "nbformat": 4,