How to use the stable-baselines.stable_baselines.common.distributions.BernoulliProbabilityDistribution function in stable-baselines

def __init__(self, logits):
        """
        Probability distributions from Bernoulli input

        :param logits: ([float]) the Bernoulli input data
        """
        self.logits = logits
        self.probabilities = tf.sigmoid(logits)
        super(BernoulliProbabilityDistribution, self).__init__()

def probability_distribution_class(self):
        return BernoulliProbabilityDistribution