Merge pull request #3 from eddelbuettel/patch-1

Update RcppArmadillo initialization
chriscarmona · Sep 9, 2022 · ad70e5f · ad70e5f
2 parents b56b962 + c9a8899
commit ad70e5f
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/get_latents.cpp b/src/get_latents.cpp
@@ -40,9 +40,8 @@ arma::mat get_latents_cpp( arma::mat Y,
   unsigned int cat_Yij;
 
   // possible variable classes that are allowed
-  arma::vec var_type_all;
-  var_type_all << 1 << 2 << 3 << arma::endr;
-
+  arma::vec var_type_all({1, 2, 3});
+
   // checking input consistency
   if( var_type.n_rows != p ) {
     throw std::range_error("The number of columns in Y have to be equal to the lenght of vector var_type");