code: def js_distance(p, q):
      
      js_dist = np.sqrt(js_divergence(p, q))
      return js_dist
docs: Compute the Jensen-Shannon distance between two discrete distributions.
  
      NOTE: JS divergence is not a metric but the sqrt of JS divergence is a
      metric and is called the JS distance.
  
      Parameters
      ----------
      p : np.array
          probability mass array (sums to 1)
      q : np.array
          probability mass array (sums to 1)
  
      Returns
      -------
      js_dist : float
          Jensen-Shannon distance between two discrete distributions