code: def callback_state(self, d):
      step = d["i"]
  
      if self.stop_at is not None and step > self.stop_at:
          raise InterruptedException
  
      state.sampling_step = step
      shared.total_tqdm.update()
  
imports: import inspect
  from collections import namedtuple
  import numpy
  import torch
  from PIL import Image
  from modules import devices, images, sd_vae_approx, sd_samplers, sd_vae_taesd, shared, sd_models
  from modules.shared import opts, state
  import k_diffusion.sampling
label_desc: Use image node_id 2 for calling the Sampler obj's underlying member method code with example usage: obj.callback_state(d) without return types