Add comment to log_partition_function