Enable rich type support by default in PyTreeMetadataOptions.

JustinPan-goog · Orbax Authors · commit 5fa970d382d9 · 2026-04-01T14:51:46.000-07:00
PiperOrigin-RevId: 893125190
diff --git a/checkpoint/orbax/checkpoint/_src/metadata/pytree_metadata_options.py b/checkpoint/orbax/checkpoint/_src/metadata/pytree_metadata_options.py
@@ -34,4 +34,4 @@ class PyTreeMetadataOptions:
 
 
 # Global default options.
-PYTREE_METADATA_OPTIONS = PyTreeMetadataOptions(support_rich_types=False)
+PYTREE_METADATA_OPTIONS = PyTreeMetadataOptions(support_rich_types=True)

Original file line number	Diff line number	Diff line change
`@@ -34,4 +34,4 @@ class PyTreeMetadataOptions:`
`34`	`34`
`35`	`35`
`36`	`36`	`# Global default options.`
`37`		`-PYTREE_METADATA_OPTIONS = PyTreeMetadataOptions(support_rich_types=False)`
	`37`	`+PYTREE_METADATA_OPTIONS = PyTreeMetadataOptions(support_rich_types=True)`