def files(self, tmp = False):
    """Return the files for the sharded embeddings.

    Args:
      tmp: Whether to return paths in /tmp or not

    Returns:
      Paths of all sharded files.
    """
    sharded_files = []
    for shard in range(self._n_shards):
      sharded_files.extend(self.shard_files(shard, tmp=tmp))

    return sharded_files
Return the files for the sharded embeddings.

    Args:
      tmp: Whether to return paths in /tmp or not

    Returns:
      Paths of all sharded files.
    
language/serene/wiki_index.py