code: def test_from_save_pretrained(self):
      kwargs = dict(self.forward_default_kwargs)
  
      num_inference_steps = kwargs.pop("num_inference_steps", None)
  
      for scheduler_class in self.scheduler_classes:
          sample, _ = self.dummy_sample
          residual = 0.1 * sample
  
          scheduler_config = self.get_scheduler_config()
          scheduler = scheduler_class(**scheduler_config)
          state = scheduler.create_state()
  
          with tempfile.TemporaryDirectory() as tmpdirname:
              scheduler.save_config(tmpdirname)
              (
                  new_scheduler,
                  new_state,
              ) = scheduler_class.from_pretrained(tmpdirname)
  
          if num_inference_steps is not None and hasattr(
              scheduler, "set_timesteps"
          ):
              state = scheduler.set_timesteps(
                  state, num_inference_steps
              )
              new_state = new_scheduler.set_timesteps(
                  new_state, num_inference_steps
              )
          elif num_inference_steps is not None and not hasattr(
              scheduler, "set_timesteps"
          ):
              kwargs["num_inference_steps"] = num_inference_steps
  
          output = scheduler.step(
              state, residual, 1, sample, **kwargs
          ).prev_sample
          new_output = new_scheduler.step(
              new_state, residual, 1, sample, **kwargs
          ).prev_sample
  
          assert (
              jnp.sum(jnp.abs(output - new_output)) < 1e-5
          ), "Scheduler outputs are not identical"
  
imports: import inspect
  import tempfile
  import unittest
  from typing import Dict, List, Tuple
  from diffusers import FlaxDDIMScheduler, FlaxDDPMScheduler, FlaxPNDMScheduler
  from diffusers.utils import is_flax_available
  from diffusers.utils.testing_utils import require_flax
label_desc: Use image node_id 4 for calling the FlaxDDIMSchedulerTest obj's underlying member method code with example usage: obj.test_from_save_pretrained() without return types