Move checksums.tsv to constants.CHECKSUMS_FILENAME

PiperOrigin-RevId: 669319279

Author: fineguy

tensorflow_datasets/core/constants.py

@@ -44,6 +44,7 @@
 DATASET_INFO_FILENAME = 'dataset_info.json'
 LICENSE_FILENAME = 'LICENSE'
 METADATA_FILENAME = 'metadata.json'
+CHECKSUMS_FILENAME = 'checksums.tsv'
 
 # Filepath for mapping between TFDS datasets and PapersWithCode entries.
 PWC_FILENAME = 'tfds_to_pwc_links.json'

tensorflow_datasets/core/dataset_builder.py

@@ -445,7 +445,7 @@ def _checksums_path(cls) -> Optional[epath.Path]:
     # * To save the checksums (in DownloadManager)
     if not cls.code_path:
       return None
-    new_path = cls.code_path.parent / "checksums.tsv"
+    new_path = cls.code_path.parent / constants.CHECKSUMS_FILENAME
     # Checksums of legacy datasets are located in a separate dir.
     legacy_path = utils.tfds_path() / "url_checksums" / f"{cls.name}.txt"
     if (

tensorflow_datasets/core/download/checksums_test.py

@@ -17,12 +17,13 @@
 
 import hashlib
 import pathlib
+from tensorflow_datasets.core import constants
 from tensorflow_datasets.core import utils
 from tensorflow_datasets.core.download import checksums
 
 
 def test_checksums(tmp_path: pathlib.Path):
-  path = tmp_path / 'checksums.tsv'
+  path = tmp_path / constants.CHECKSUMS_FILENAME
   url_infos = {
       'http://abc.org/data': checksums.UrlInfo(
           checksum='abcd',

tensorflow_datasets/scripts/cli/new.py

@@ -211,8 +211,8 @@ def _create_dummy_data(info: utils.DatasetInfo) -> None:
 
 
 def _create_checksum(info: utils.DatasetInfo) -> None:
-  """Adds the `checksums.tsv` file."""
-  file_path = info.path / 'checksums.tsv'
+  """Adds the checksum file."""
+  file_path = info.path / constants.CHECKSUMS_FILENAME
   content = textwrap.dedent(f"""\
       # {info.todo}: If your dataset downloads files, then the checksums
       # will be automatically added here when running