From 232ce2a3425cebd672c64bb43cfff0d5f84ece28 Mon Sep 17 00:00:00 2001
From: Ryan Curtin <ryan@ratml.org>
Date: Sat, 2 Apr 2022 15:37:55 -0400
Subject: [PATCH] Adapt CustomLayer for testing to have only one template
 parameter.

---
 src/mlpack/tests/custom_layer.hpp | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/src/mlpack/tests/custom_layer.hpp b/src/mlpack/tests/custom_layer.hpp
index 8d791b1ebd..3a3eaa78ad 100644
--- a/src/mlpack/tests/custom_layer.hpp
+++ b/src/mlpack/tests/custom_layer.hpp
@@ -26,11 +26,9 @@ namespace ann {
  */
 template <
     class ActivationFunction = LogisticFunction,
-    typename InputDataType = arma::mat,
-    typename OutputDataType = arma::mat
+    typename MatType = arma::mat
 >
-using CustomLayer = BaseLayer<
-    ActivationFunction, InputDataType, OutputDataType>;
+using CustomLayer = BaseLayer<ActivationFunction, MatType>;
 
 } // namespace ann
 } // namespace mlpack