tfds-nightly 4.9.9.dev202508050045__py3-none-any.whl → 4.9.9.dev202508070045__py3-none-any.whl

tensorflow_datasets/core/dataset_info.py

@@ -719,7 +719,7 @@ class DatasetInfo:
       )
 
     self._identity = DatasetIdentity.from_proto(
-        info_proto=parsed_proto, data_dir=dataset_info_dir
+        info_proto=parsed_proto, data_dir=os.fspath(dataset_info_dir)
     )
 
     # Update splits

tensorflow_datasets/scripts/cli/croissant.py

@@ -125,16 +125,11 @@ class CmdArgs(simple_parsing.helpers.FrozenSerializable):
 
 def register_subparser(parsers: argparse._SubParsersAction):
   """Add subparser for `convert_format` command."""
-  orig_parser_class = parsers._parser_class  # pylint: disable=protected-access
-  try:
-    parsers._parser_class = simple_parsing.ArgumentParser  # pylint: disable=protected-access
-    parser = parsers.add_parser(
-        'build_croissant',
-        help='Prepares a croissant dataset',
-    )
-    parser = typing.cast(simple_parsing.ArgumentParser, parser)
-  finally:
-    parsers._parser_class = orig_parser_class  # pylint: disable=protected-access
+  parser = parsers.add_parser(
+      'build_croissant',
+      help='Prepares a croissant dataset',
+  )
+  parser = typing.cast(simple_parsing.ArgumentParser, parser)
   parser.add_arguments(CmdArgs, dest='args')
   parser.set_defaults(
       subparser_fn=lambda args: prepare_croissant_builders(args.args)

tensorflow_datasets/scripts/cli/main.py

@@ -28,8 +28,8 @@ from typing import List
 from absl import app
 from absl import flags
 from absl import logging
-from absl.flags import argparse_flags
 
+import simple_parsing
 import tensorflow_datasets.public_api as tfds
 
 # Import commands
@@ -46,7 +46,7 @@ def _parse_flags(argv: List[str]) -> argparse.Namespace:
   """Command lines flag parsing."""
   argv = flag_utils.normalize_flags(argv)  # See b/174043007 for context.
 
-  parser = argparse_flags.ArgumentParser(
+  parser = simple_parsing.ArgumentParser(
       description='Tensorflow Datasets CLI tool',
       allow_abbrev=False,
   )
@@ -67,7 +67,22 @@ def _parse_flags(argv: List[str]) -> argparse.Namespace:
   new.register_subparser(subparser)
   convert_format.register_subparser(subparser)
   croissant.register_subparser(subparser)
-  return parser.parse_args(argv[1:])
+
+  namespace, remaining_argv = parser.parse_known_args(argv[1:])
+
+  # Manually parse absl flags from the remaining arguments.
+  try:
+    # FLAGS requires the program name as the first argument.
+    positionals = FLAGS(argv[:1] + remaining_argv)
+  except flags.Error as e:
+    parser.error(str(e))
+
+  # There should be no positional arguments left, as they should have been
+  # handled by the sub-commands.
+  if len(positionals) > 1:
+    parser.error(f"unrecognized arguments: {' '.join(positionals[1:])}")
+
+  return namespace
 
 
 def main(args: argparse.Namespace) -> None:

{tfds_nightly-4.9.9.dev202508050045.dist-info → tfds_nightly-4.9.9.dev202508070045.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tfds-nightly
-Version: 4.9.9.dev202508050045
+Version: 4.9.9.dev202508070045
 Summary: tensorflow/datasets is a library of datasets ready to use with TensorFlow.
 Home-page: https://github.com/tensorflow/datasets
 Download-URL: https://github.com/tensorflow/datasets/tags

{tfds_nightly-4.9.9.dev202508050045.dist-info → tfds_nightly-4.9.9.dev202508070045.dist-info}/RECORD

@@ -67,7 +67,7 @@ tensorflow_datasets/core/dataset_builder_read_test.py,sha256=QxodggixId7zmknhnC0
 tensorflow_datasets/core/dataset_builder_test.py,sha256=Q2GlADkgd0hzHHPLdMalW1tyc00QPdPNzaLxR5oU4HQ,49257
 tensorflow_datasets/core/dataset_collection_builder.py,sha256=9tvIWFL4gLFaDFnU_ioKaGGzPQshmACS64vagjTVzkE,7224
 tensorflow_datasets/core/dataset_collection_builder_test.py,sha256=m6H9Kxb-RHUiBEZXd6kyheoixlogJ3OwzixHM2jP6WA,3664
-tensorflow_datasets/core/dataset_info.py,sha256=NHerzNwv9L_ZUE1vvd9O9pS06Si7rCXGYaiTiUF1PJY,52625
+tensorflow_datasets/core/dataset_info.py,sha256=jtkzrc0Jw5BnLnbR18hvPwlzzj-86Rib_9eHlAehYa8,52636
 tensorflow_datasets/core/dataset_info_test.py,sha256=T9wnE8XMf1Cl_5IX6fx55AHp7q40mTTQG9H_dyBlTLM,31522
 tensorflow_datasets/core/dataset_metadata.py,sha256=ffmBbfrwsiBqMfwrbzm2EcpmmzmEhIUJNuTkz3oZW3U,3091
 tensorflow_datasets/core/dataset_metadata_test.py,sha256=gbY2Yv1Q8_euuZ0DGaZu3g-0KteqoYoTwrBgyf-uFbc,1822
@@ -1988,8 +1988,8 @@ tensorflow_datasets/scripts/cli/conftest.py,sha256=cmvCCV-efT5ZXYPkCSGS1OxoKNPAf
 tensorflow_datasets/scripts/cli/convert_format.py,sha256=Cefg7Cd4Nmz8AlwEm68T286YEuoDHtRfQfdtq7XAZrA,4805
 tensorflow_datasets/scripts/cli/convert_format_utils.py,sha256=U_q5WVgMNrjBkOc166U4Y_eca5KOS3Xb3jSDjp4XdK4,29078
 tensorflow_datasets/scripts/cli/convert_format_utils_test.py,sha256=9JGNu9TvUWzbuhe6DWwnO3V9Lia5S1Is64re-pceAWE,8823
-tensorflow_datasets/scripts/cli/croissant.py,sha256=CMOe-8iEN7ZdRTiJ5w3iiJFYKsHiqzroAXmbfJimp1Y,6396
-tensorflow_datasets/scripts/cli/main.py,sha256=cDlnkHU2-DIjFmMQo3gGPJ7pVwGY48ypqR1xm5E82TU,3910
+tensorflow_datasets/scripts/cli/croissant.py,sha256=6jzmOXt_i7aeJHUVX7_zpRRMEXId_PzU24zUDdExRUs,6112
+tensorflow_datasets/scripts/cli/main.py,sha256=FJJwyUtM1N9gNDsxGm850m5ejzzJ9mgESNW9Xz8E9_I,4383
 tensorflow_datasets/scripts/cli/main_test.py,sha256=3zNaS_2FmxxLoZOX05iJ2riuP4Qv8cx6bhAI56tV8YI,1067
 tensorflow_datasets/scripts/cli/new.py,sha256=eTAEQ5QVj7DZwyymsxNW-5KNRJb-YYWIUGcW7yhC2Ew,7699
 tensorflow_datasets/scripts/cli/new_test.py,sha256=USr9So-FPtg8UzaQPPacXn0E1ukDIoew9oYkOn45oik,2655
@@ -2468,10 +2468,10 @@ tensorflow_datasets/vision_language/wit/wit_test.py,sha256=PXS8DMNW-MDrT2p5oy4Ic
 tensorflow_datasets/vision_language/wit_kaggle/__init__.py,sha256=vGwSGeM8WE4Q-l0-eEE1sBojmk6YT0l1OO60AWa4Q40,719
 tensorflow_datasets/vision_language/wit_kaggle/wit_kaggle.py,sha256=q-vX_FBzIwsFxL4sY9vuyQ3UQD2PLM4yhUR4U6l-qao,16903
 tensorflow_datasets/vision_language/wit_kaggle/wit_kaggle_test.py,sha256=ZymHT1NkmD-pUnh3BmM3_g30c5afsWYnmqDD9dVyDSA,1778
-tfds_nightly-4.9.9.dev202508050045.dist-info/licenses/AUTHORS,sha256=nvBG4WwfgjuOu1oZkuQKw9kg7X6rve679ObS-YDDmXg,309
-tfds_nightly-4.9.9.dev202508050045.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-tfds_nightly-4.9.9.dev202508050045.dist-info/METADATA,sha256=JY3gR2KiCJReRpbWjIvTlSWW7XLRr7HBUduWEJr-Whc,11694
-tfds_nightly-4.9.9.dev202508050045.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-tfds_nightly-4.9.9.dev202508050045.dist-info/entry_points.txt,sha256=eHEL7nF5y1uCY2FgkuYIdE062epJXlAQTSdq89px4p4,73
-tfds_nightly-4.9.9.dev202508050045.dist-info/top_level.txt,sha256=bAevmk9209s_oxVZVlN6hSDIVS423qrMQvmcWSvW4do,20
-tfds_nightly-4.9.9.dev202508050045.dist-info/RECORD,,
+tfds_nightly-4.9.9.dev202508070045.dist-info/licenses/AUTHORS,sha256=nvBG4WwfgjuOu1oZkuQKw9kg7X6rve679ObS-YDDmXg,309
+tfds_nightly-4.9.9.dev202508070045.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+tfds_nightly-4.9.9.dev202508070045.dist-info/METADATA,sha256=tvgIHPVBd-5pYbRiVmundzy13s1KUvUJsbwFeEktwX8,11694
+tfds_nightly-4.9.9.dev202508070045.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+tfds_nightly-4.9.9.dev202508070045.dist-info/entry_points.txt,sha256=eHEL7nF5y1uCY2FgkuYIdE062epJXlAQTSdq89px4p4,73
+tfds_nightly-4.9.9.dev202508070045.dist-info/top_level.txt,sha256=bAevmk9209s_oxVZVlN6hSDIVS423qrMQvmcWSvW4do,20
+tfds_nightly-4.9.9.dev202508070045.dist-info/RECORD,,