Only passing federated-plugin parameter to plugins (dmlc#10629)

ZiyueXu77 · Jul 23, 2024 · be86d0e · be86d0e
1 parent c26c122
commit be86d0e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/plugin/federated/federated_plugin.cc b/plugin/federated/federated_plugin.cc
@@ -125,7 +125,7 @@ FederatedPlugin::FederatedPlugin(StringView path, Json config)
   plugin_handle_ = decltype(plugin_handle_)(
       [&] {
         // Initialize the parameters
-        auto const& obj = get<Object>(config);
+        auto const& obj = OptionalArg<Object>(config, "federated_plugin", Object::Map{});
         std::vector<std::string> kwargs;
         for (auto const& kv : obj) {
           std::string value;