def __init__()

in maddpg/common/distributions.py [0:0]

• 4 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(self, low, high, flat):
        self.flat = flat
        self.low = tf.constant(low, dtype=tf.float32)
        self.categoricals = list(map(SoftCategoricalPd, tf.split(flat, high - low + 1, axis=len(flat.get_shape()) - 1)))