Make Croissant CmdArgs frozen and serializable.

fineguy · The TensorFlow Datasets Authors · commit 8ab19aa814df · 2024-06-12T09:00:31.000-07:00
PiperOrigin-RevId: 642630303
diff --git a/tensorflow_datasets/scripts/cli/croissant.py b/tensorflow_datasets/scripts/cli/croissant.py
@@ -38,8 +38,8 @@
 from tensorflow_datasets.scripts.cli import cli_utils
 
 
-@dataclasses.dataclass
-class CmdArgs:
+@dataclasses.dataclass(frozen=True, kw_only=True)
+class CmdArgs(simple_parsing.helpers.FrozenSerializable):
   """CLI arguments for preparing a Croissant dataset.
 
   Attributes:
