def _preprocess_parameters_for_log_prob(self, x, **parameters): tuple_x, parameters = broadcast_and_squeeze_mixed( tuple([x]), parameters) return tuple_x[ 0], parameters, None, None #TODO: add proper output here
def _preprocess_parameters_for_sampling(self, **parameters): parameters = broadcast_and_squeeze_mixed((), parameters) return parameters, None