Fixed PopularityBasedSamplerV2 that was raising error when the sampled item is equal the item id cardinality

Author: gabrielspmoreira

merlin/models/tf/outputs/sampling/popularity.py

@@ -147,7 +147,7 @@ def get_sampling_distribution(self) -> tf.Tensor:
         tf.Tensor
             Probabilities of each item to be sampled
         """
-        log_indices = tf.math.log(tf.range(1.0, self.max_id - self.min_id + 2.0, 1.0))
+        log_indices = tf.math.log(tf.range(1.0, self.max_id - self.min_id + 3.0, 1.0))
         sampling_probs = (log_indices[1:] - log_indices[:-1]) / log_indices[-1]
 
         if self.unique: