def distributed_maybe_download()

in train_helpers.py [0:0]

• 7 lines of code
• 2 McCabe index (conditional complexity)

def distributed_maybe_download(path, local_rank, mpi_size):
    if not path.startswith('gs://'):
        return path
    filename = path[5:].replace('/', '-')
    with first_rank_first(local_rank, mpi_size):
        fp = maybe_download(path, filename)
    return fp