code: def __init__(self, diffusion):
      self.diffusion = diffusion
      self._weights = np.ones([diffusion.num_timesteps])
  
imports: from abc import ABC, abstractmethod
  import numpy
  import torch
  from scipy.stats import norm
  import torch.distributed
label_desc: Use image node_id 1 to create a new UniformSampler object from inherited base classes: ScheduleSampler with example: obj = UniformSampler(diffusion)