Reject duplicate input names in C Predict API

apache · Sep 20, 2022 · 26fa682 · 26fa682
1 parent e2ed553
commit 26fa682
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/c_api/c_predict_api.cc b/src/c_api/c_predict_api.cc
@@ -201,7 +201,9 @@ int _CreatePartialOut(const char* symbol_json_str,
   std::unordered_map<std::string, size_t> key2arg;
   for (size_t i = 0; i < arg_names.size(); ++i) {
     std::string key = arg_names[i];
-    key2arg[key] = i;
+    auto emplace_res = ret->key2arg.emplace(key, i);
+    CHECK(emplace_res.second)
+      << "Input parameter name appears more than once";
   }
 
   try {