code: def create(
      cls,
      common: CommonSchedulerState,
      final_alpha_cumprod: jnp.ndarray,
      init_noise_sigma: jnp.ndarray,
      timesteps: jnp.ndarray,
  ):
      return cls(
          common=common,
          final_alpha_cumprod=final_alpha_cumprod,
          init_noise_sigma=init_noise_sigma,
          timesteps=timesteps,
      )
  
imports: from dataclasses import dataclass
  from typing import Optional, Tuple, Union
  import flax
  import jax
  import jax.numpy
  from ..configuration_utils import ConfigMixin, register_to_config
  from .scheduling_utils_flax import CommonSchedulerState, FlaxKarrasDiffusionSchedulers, FlaxSchedulerMixin, FlaxSchedulerOutput, add_noise_common
label_desc: Use image node_id 1 for calling the PNDMSchedulerState obj's underlying member method code with example usage: obj.create(cls, common, final_alpha_cumprod, init_noise_sigma, timesteps) and returns: cls