From 2acea2db66aaab9725abcdcac389b8376cc3dc04 Mon Sep 17 00:00:00 2001
From: Ryan Curtin <ryan@ratml.org>
Date: Wed, 29 Nov 2023 11:11:25 -0500
Subject: [PATCH] Fix bug in Naive Bayes test.

---
 src/mlpack/tests/nbc_test.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/mlpack/tests/nbc_test.cpp b/src/mlpack/tests/nbc_test.cpp
index 56df32b762..626594f266 100644
--- a/src/mlpack/tests/nbc_test.cpp
+++ b/src/mlpack/tests/nbc_test.cpp
@@ -457,7 +457,7 @@ TEMPLATE_TEST_CASE("NBCIncrementalTest", "[NBCTest]", arma::fmat, arma::mat)
     labels[i] = trainData(trainData.n_rows - 1, i);
   trainData.shed_row(trainData.n_rows - 1);
 
-  NaiveBayesClassifier<MatType> nbc(trainData.n_rows, classes);
+  NaiveBayesClassifier<MatType> nbc(trainData.n_rows, 2);
 
   for (size_t i = 0; i < trainData.n_cols; ++i)
   {