chop.passes.interface.save_and_load

chop.passes.interface.save_and_load#

load_mase_graph_interface_pass#

chop.passes.graph.interface.save_and_load.load_mase_graph_interface_pass(graph, pass_args: dict = {'load_dir': None})[source]#

Load the MASE graph interface pass.

Parameters:

  • graph (MaseGraph) – The input graph to be transformed.

  • pass_args (dict) – Optional arguments for the transformation pass. Default is {‘load_dir’: None}, load_dir is required.

Returns:

The transformed graph and an empty dictionary.

Return type:

tuple(MaseGraph, dic)

Raises:

ValueError – If the load directory is not specified.

save_mase_graph_interface_pass#

chop.passes.graph.interface.save_and_load.save_mase_graph_interface_pass(graph, pass_args: dict = {})[source]#

Save a mase graph.

This saves the graph module as a serialized graph module and metadata.parameters as a toml file.

Parameters:

  • graph (MaseGraph) – mase_graph to save

  • pass_args (str) – save directory

Returns:

mase_graph

Return type:

MaseGraph

load_node_graph_interface_pass#

chop.passes.graph.interface.load_node_meta_param_interface_pass(graph, pass_args: str)[source]#

Load a mase graph metadata.parameters from a toml file.

save_node_graph_interface_pass#

chop.passes.graph.interface.save_node_meta_param_interface_pass(graph, pass_args: str)[source]#

Save a mase graph metadata.parameters to a toml file.