From 5e508df704df2cb795e4123f777e5f558ff992b9 Mon Sep 17 00:00:00 2001
From: lancer <tangshao28@gmail.com>
Date: Sun, 14 Jan 2024 20:43:57 -0800
Subject: [PATCH] Ensure the constness of the data member

---
 cpp/custom-dataset/custom-dataset.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/cpp/custom-dataset/custom-dataset.cpp b/cpp/custom-dataset/custom-dataset.cpp
index d4ad51af22..a9f3462f48 100644
--- a/cpp/custom-dataset/custom-dataset.cpp
+++ b/cpp/custom-dataset/custom-dataset.cpp
@@ -26,7 +26,7 @@ using Data = std::vector<std::pair<std::string, long>>;
 class CustomDataset : public torch::data::datasets::Dataset<CustomDataset> {
   using Example = torch::data::Example<>;
 
-  Data data;
+  const Data data;
 
  public:
   CustomDataset(const Data& data) : data(data) {}
@@ -198,7 +198,7 @@ int main() {
   std::cout << "Running on: "
             << (options.device == torch::kCUDA ? "CUDA" : "CPU") << std::endl;
 
-  auto data = readInfo();
+  const auto data = readInfo();
 
   auto train_set =
       CustomDataset(data.first).map(torch::data::transforms::Stack<>());