From e48e4b7e810dc644759d19543974717759cb2f57 Mon Sep 17 00:00:00 2001
From: Ryan Curtin <ryan@ratml.org>
Date: Wed, 16 Jun 2021 14:24:42 -0400
Subject: [PATCH] Use SetDimensionality(); the constructor will lose the
 policy.

---
 src/mlpack/core/data/load_csv.hpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/mlpack/core/data/load_csv.hpp b/src/mlpack/core/data/load_csv.hpp
index 845d3f076f..ce7d0bb9b1 100644
--- a/src/mlpack/core/data/load_csv.hpp
+++ b/src/mlpack/core/data/load_csv.hpp
@@ -100,7 +100,7 @@ class LoadCSV
     // Reset the DatasetInfo object, if needed.
     if (info.Dimensionality() == 0)
     {
-      info = DatasetMapper<MapPolicy>(rows);
+      info.SetDimensionality(rows);
     }
     else if (info.Dimensionality() != rows)
     {
@@ -195,7 +195,7 @@ class LoadCSV
         // Reset the DatasetInfo object, if needed.
         if (info.Dimensionality() == 0)
         {
-          info = DatasetMapper<MapPolicy>(rows);
+          info.SetDimensionality(rows);
         }
         else if (info.Dimensionality() != rows)
         {