Ensure the const-ness of the data member in cpp/custom-dataset (#1215)

pytorch · Jan 23, 2024 · b2832cc · b2832cc
1 parent 42b3bda
commit b2832cc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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<>());