def __init__(self, *, embedding_dir, n_shards):
    """Initialize sharded paths.

    Args:
      embedding_dir: Directory of embeddings
      n_shards: Number of shards in directory
    """
    self._embedding_dir = embedding_dir
    self._n_shards = n_shards
    self._tmp_dir = os.path.join('/tmp',
                                 util.random_string(prefix='embedding-tmp'))
Initialize sharded paths.

    Args:
      embedding_dir: Directory of embeddings
      n_shards: Number of shards in directory
    
language/serene/wiki_index.py