8af731463df46f6c73c933bdc45ed9a4ecd4e422,nni/retiarii/codegen/pytorch.py,,model_to_pytorch_script,#Any#,7

Before Change

def model_to_pytorch_script(model: Model) -> str:
    graphs = [graph_to_pytorch_model(name, cell) for name, cell in model.graphs.items()]
    return _PyTorchScriptTemplate.format("\n\n".join(graphs)).strip()


def _sorted_incoming_edges(node: Node) -> List[Edge]:

After Change

        graphs.append(graph_code)
        total_pkgs.update(import_pkgs)
    // TODO: set correct PATH for the packages (after launch refactor)
    pkgs_code = "\n".join(["import {}".format(pkg) for pkg in total_pkgs])
    return _PyTorchScriptTemplate.format(pkgs_code, "\n\n".join(graphs)).strip()

def _convert_name(name: str) -> str:

Instances