[c++][fix] check nullable of bin mappers in dataset_loader.cpp (fix m…

…icrosoft#5221) (microsoft#5258) check nullable of bin mappers
jameslamb · Jun 2, 2022 · fa9e452 · fa9e452
1 parent 4971a06
commit fa9e452
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/io/dataset_loader.cpp b/src/io/dataset_loader.cpp
@@ -1472,7 +1472,7 @@ void DatasetLoader::CheckCategoricalFeatureNumBin(
   if (bin_mappers.size() < 1024) {
     for (size_t i = 0; i < bin_mappers.size(); ++i) {
       const int max_bin_for_this_feature = max_bin_by_feature.empty() ? max_bin : max_bin_by_feature[i];
-      if (bin_mappers[i]->bin_type() == BinType::CategoricalBin && bin_mappers[i]->num_bin() > max_bin_for_this_feature) {
+      if (bin_mappers[i] != nullptr && bin_mappers[i]->bin_type() == BinType::CategoricalBin && bin_mappers[i]->num_bin() > max_bin_for_this_feature) {
         need_warning = true;
         break;
       }