Update expected error message in save_load_test.py.

PiperOrigin-RevId: 930323519

Author: abhishek002002

checkpoint/orbax/checkpoint/_src/testing/oss/tagged_tests_presubmit.yaml

@@ -13,6 +13,8 @@ processes:1:
 - orbax/checkpoint/experimental/emergency/multi_tier_checkpointing:pathways_replicator_checkpoint_manager_test
 - orbax/checkpoint:single_host_test
 processes:2:
+- orbax/checkpoint/_src/checkpointers:async_checkpointer_test
+- orbax/checkpoint/_src/checkpointers:checkpointer_test
 - orbax/checkpoint/_src/handlers:array_checkpoint_handler_test
 - orbax/checkpoint/_src/handlers:pytree_checkpoint_handler_test
 - orbax/checkpoint/_src/handlers:standard_checkpoint_handler_test
@@ -21,6 +23,18 @@ processes:2:
 - orbax/checkpoint/experimental/emergency/p2p:checkpoint_manager_multiprocess_test
 - orbax/checkpoint/experimental/emergency/p2p:local_multiprocess_test
 - orbax/checkpoint/experimental/emergency/p2p:persistent_multiprocess_test
+- orbax/checkpoint/experimental/v1/_src/handlers:pytree_handler_partial_save_test
+- orbax/checkpoint/experimental/v1/_src/handlers:pytree_handler_test
+- orbax/checkpoint/experimental/v1/_src/layout:orbax_layout_multiprocess_test
+- orbax/checkpoint/experimental/v1/_src/partial:saving_multihost_test
+- orbax/checkpoint/experimental/v1/_src/serialization:array_leaf_handler_test
+- orbax/checkpoint/experimental/v1/_src/serialization:numpy_leaf_handler_test
+- orbax/checkpoint/experimental/v1/_src/serialization:scalar_leaf_handler_test
+- orbax/checkpoint/experimental/v1/_src/serialization:string_leaf_handler_test
+- orbax/checkpoint/experimental/v1/_src/testing:save_load_test
+- orbax/checkpoint/experimental/v1/_src/testing:v1_compatibility_load_test_multiprocess
+- orbax/checkpoint/experimental/v1/_src/training:checkpointer_test
+- orbax/checkpoint/experimental/v1/_src/training:v0v1_compatibility_checkpointer_test
 processes:4:
 - orbax/checkpoint/_src/multihost:multihost_test
 - orbax/checkpoint/_src/testing/tree_verity:checkpoint_manager_test

checkpoint/orbax/checkpoint/experimental/v1/_src/testing/save_load_test.py

@@ -350,9 +350,7 @@ def test_save_unregistered_type(self):
         expected_msg = (
             'Unsupported leaf type for saving:'
             f' `{handler_utils.Foo}`.'
-            ' Supported leaf types are:'
-            ' jax.Array | numpy.ndarray | int | float | numpy.number | bytes |'
-            ' bool | str.'
+            f' Supported leaf types are: {tree_types.Leaf}.'
         )
         with self.assertRaisesRegex(ValueError, re.escape(expected_msg)):
           ocp.save(self.directory, handler_utils.Foo(1, 'hi'))