diff --git a/aidge_core/show_graphview.py b/aidge_core/show_graphview.py index 633298f10dbfdafe40022f88f741f82d2d35c681..23d4b5f370056d0fde3a3d937fd2f0da0a5351ad 100644 --- a/aidge_core/show_graphview.py +++ b/aidge_core/show_graphview.py @@ -129,7 +129,7 @@ def _create_dict(ordered_nodes : list[aidge_core.Node], write_trainable_params_e if params_file_format=='npz': np.savez_compressed(Path(path_trainable_params, node.name()), **{node.name() : node.get_operator().get_output(0)}) - node_dict['tensor_data'] = Path(path_trainable_params, node.name() + '.npz') + node_dict['tensor_data'] = str(Path(path_trainable_params, node.name() + '.npz')) elif params_file_format=='json': tensor = np.array(node.get_operator().get_output(0)) @@ -145,7 +145,7 @@ def _create_dict(ordered_nodes : list[aidge_core.Node], write_trainable_params_e with open(Path(path_trainable_params, node.name() + '.json'), 'w') as fp: json.dump(tensor_dict, fp, indent=4) - node_dict['tensor_data'] = Path(path_trainable_params, node.name() + '.json') + node_dict['tensor_data'] = str(Path(path_trainable_params, node.name() + '.json')) else: raise Exception("File format to write trainable parameters not recognized.")