magic.py 文件源码

def run_in_cloud(cell, connector, namespace):

    local_vars = get_local_vars(cell, namespace)
    imports, unused_vars = find_required_imports(cell, local_vars)

    for var in unused_vars:
        del local_vars[var]

    uploads = {}
    uploads['cell'] = cell
    uploads['env'] = local_vars
    uploads['imports'] = imports

    with open('uploads.pkl', 'wb') as file:
        dill.dump(uploads, file)

    server_info = connector.contact_server()
    if (server_info is None):
        return 

    gpu_hash, gpu_ip, ws_port = server_info

    connector.upload_params_magic(gpu_ip, gpu_hash)
    outUrl = connector.stream_output(gpu_ip, gpu_hash, ws_port)

    if outUrl is None:
        return 

    result = connector.get_return_object(outUrl)
    return result