    def test_epoch_end(self, outputs):
        """
        Computes average test accuracy score

        :param outputs: outputs after every epoch end

        :return: output - average test loss
        """
        avg_test_acc = torch.stack([x["test_acc"] for x in outputs]).mean()
        self.log("avg_test_acc", avg_test_acc, sync_dist=True)
Computes average test accuracy score

:param outputs: outputs after every epoch end

:return: output - average test loss
tests/integration/frameworks/mlflow/MNIST/mnist_autolog_example.py