class_docstr: Shuffles / writes Examples beam collection to sharded TFRecord files.
  
  The given examples are not directly writen to the final tfrecord file, but to
  temporary files.
class_docstr_tok: ["Shuffles","\/","writes","Examples","beam","collection","to","sharded","TFRecord","files",".","The","given","examples","are","not","directly","writen","to","the","final","tfrecord","file",",","but","to","temporary","files","."]
docstr: Stores the split info to disk.
docstr_tok: ["Stores","the","split","info","to","disk","."]
code: def _store_split_info(
      self,
      shard_infos: List[_ShardInfo],
      total_size: int,
  ) -> None:
      """Stores the split info to disk."""
      shard_infos = sorted(shard_infos, key=lambda x: x.id)
      shard_lengths = [info.num_examples for info in shard_infos]
      with utils.incomplete_file(
          epath.Path(self._split_info_path)
      ) as tmp_path:
          tmp_path.write_text(
              json.dumps(
                  {
                      "total_size": total_size,
                      "shard_lengths": shard_lengths,
                  }
              )
          )
  
imports: import dataclasses
  import functools
  import itertools
  import json
  import os
  from typing import Any, Iterable, List, Optional, Tuple
  from absl import logging
  from etils import epath
  from tensorflow_datasets.core import example_parser
  from tensorflow_datasets.core import example_serializer
  from tensorflow_datasets.core import file_adapters
  from tensorflow_datasets.core import hashing
  from tensorflow_datasets.core import lazy_imports_lib
  from tensorflow_datasets.core import naming
  from tensorflow_datasets.core import shuffle
  from tensorflow_datasets.core import utils
  from tensorflow_datasets.core.utils import file_utils
  from tensorflow_datasets.core.utils import shard_utils
  from tensorflow_datasets.core.utils import type_utils
  from tensorflow_datasets.core.utils.lazy_imports_utils import apache_beam
label_desc: Use image node_id 12 for calling the BeamWriter obj's underlying member method code with example usage: obj._store_split_info(shard_infos, total_size) without return types