def restore_file()

in coinrun/setup_utils.py [0:0]

• 17 lines of code
• 4 McCabe index (conditional complexity)

def restore_file(restore_id, load_key='default'):
    if restore_id is not None:
        load_file = Config.get_load_filename(restore_id=restore_id)
        filepath = file_to_path(load_file)
        load_data = joblib.load(filepath)

        Config.set_load_data(load_data, load_key=load_key)

        restored_args = load_data['args']
        sub_dict = {}
        res_keys = Config.RES_KEYS

        for key in res_keys:
            if key in restored_args:
                sub_dict[key] = restored_args[key]
            else:
                print('warning key %s not restored' % key)

        Config.parse_args_dict(sub_dict)
    
    from coinrun.coinrunenv import init_args_and_threads
    init_args_and_threads(4)