def _do_save(save_op_tuple: Tuple): set_io_scopes(caller_io_scopes) if len(save_op_tuple) == 2: return save(save_op_tuple[0], save_op_tuple[1], save_context=current_save_context) elif len(save_op_tuple) == 3: return save(save_op_tuple[0], save_op_tuple[1], save_context=current_save_context, **(save_op_tuple[2])) else: raise ValueError(f'unknown save tuple size: {len(save_op_tuple)}')
def test_set_write_scopes(): target_path = "./tests/fixtures/write_scope5.txt" if os.path.exists(target_path): os.remove(target_path) # fake write scopes, such as when called on a remote worker: _old_scopes = scoping.current_io_scopes() scoping.set_io_scopes(["./tests", "fixtures"]) write("test", "write_scope5.txt", mode="w") assert os.path.isfile(target_path) # restore write scopes for later tests scoping.set_io_scopes(_old_scopes)
def _do_load(url): set_io_scopes(caller_io_scopes) return load(url, cache=cache, **kwargs)