From 2f71e925d00d23ad39ed28b86ee3a95170481e8a Mon Sep 17 00:00:00 2001
From: Omar Shrit <omar@avontech.fr>
Date: Thu, 21 Mar 2024 07:11:58 +0100
Subject: [PATCH] Remove fixing the dimension manually

Signed-off-by: Omar Shrit <omar@avontech.fr>
---
 src/mlpack/tests/ann/layer/parametric_relu.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/mlpack/tests/ann/layer/parametric_relu.cpp b/src/mlpack/tests/ann/layer/parametric_relu.cpp
index 35d8dd037b..c300050087 100644
--- a/src/mlpack/tests/ann/layer/parametric_relu.cpp
+++ b/src/mlpack/tests/ann/layer/parametric_relu.cpp
@@ -117,7 +117,7 @@ TEST_CASE("PReLUIntegrationTest", "[ANNLayerTest]")
   model.Add<Linear>(3);
   model.Add<PReLU>(0.01);
   model.Add<Linear>(1);
-  model.InputDimensions() = std::vector<size_t>({trainData.n_rows});
+
   // Sometimes the model may not optimize correctly, so we allow a few trials.
   bool success = false;
   for (size_t trial = 0; trial < 3; ++trial)