import resource


def fix_ulimit():
    want_rlimit = 524288
    rlimit = resource.getrlimit(resource.RLIMIT_NOFILE)
    if want_rlimit > rlimit[1]:
        raise RuntimeError(
            f"Can't increase RLIMIT_NOFILE - want {want_rlimit} but hard limit is {rlimit[1]}"
        )
    resource.setrlimit(resource.RLIMIT_NOFILE, (want_rlimit, rlimit[1]))
