code: def draw_flat_samples(**kwargs):
      ''' Draw samples for uniform in mass
  
          Parameters
          ----------
          **kwargs: string
             Keyword arguments as model parameters and number of samples
  
          Returns
          -------
          array
             The first mass
          array
             The second mass
      '''
  
      #PDF doesnt match with sampler
      nsamples = kwargs.get('nsamples', 1)
      min_mass = kwargs.get('min_mass', 1.)
      max_mass = kwargs.get('max_mass', 2.)
  
      m1 = np.random.uniform(min_mass, max_mass, nsamples)
      m2 = np.random.uniform(min_mass, max_mass, nsamples)
  
      return np.maximum(m1, m2), np.minimum(m1, m2f draw_flat_samples(**kwargs):
      ''' Draw samples for uniform in mass
  
          Parameters
          ----------
          **kwargs: string
             Keyword arguments as model parameters and number of samples
  
          Returns
          -------
          array
             The first mass
          array
             The second mass
      '''
  
      #PDF doesnt match with sampler
      nsamples = kwargs.get('nsamples', 1)
      min_mass = kwargs.get('min_mass', 1.)
      max_mass = kwargs.get('max_mass', 2.)
  
      m1 = np.random.uniform(min_mass, max_mass, nsamples)
      m2 = np.random.uniform(min_mass, max_mass, nsamples)
  
      return np.maximum(m1, m2), np.minimum(m1, m2)
docs: Draw samples for uniform in mass
  
          Parameters
          ----------
          **kwargs: string
             Keyword arguments as model parameters and number of samples
  
          Returns
          -------
          array
             The first mass
          array
             The second mass