code: def test_registered_saver_fails_in_saved_model_graph_mode(self):
      with context.eager_mode():
          p1 = Part([1, 4])
          p2 = Part([2, 5])
          p3 = Part([3, 6])
          s = Stack([p1, p2, p3])
          save_dir = os.path.join(self.get_temp_dir(), "save_dir")
          save.save(s, save_dir)
  
      with self.assertRaisesRegex(
          NotImplementedError,
          "registered checkpoint saver is not supported in graph mode",
      ):
          load.load(save_dir)
  
imports: import os
  import tempfile
  from absl.testing import parameterized
  from google.protobuf import wrappers_pb2
  from tensorflow.python.checkpoint import checkpoint
  from tensorflow.python.client import session
  from tensorflow.python.eager import context
  from tensorflow.python.eager import def_function
  from tensorflow.python.eager import test
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import errors_impl
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import array_ops_stack
  from tensorflow.python.ops import io_ops
  from tensorflow.python.ops import resource_variable_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import gfile
  from tensorflow.python.saved_model import load
  from tensorflow.python.saved_model import loader
  from tensorflow.python.saved_model import registration
  from tensorflow.python.saved_model import save
  from tensorflow.python.trackable import autotrackable
label_desc: Use image node_id 1 for calling the SingleCycleTest obj's underlying member method code with example usage: obj.test_registered_saver_fails_in_saved_model_graph_mode() without return types