diff --git a/plugin/federated/engine_federated.cc b/plugin/federated/engine_federated.cc index 8fd396d94744..9b43c3997cc3 100644 --- a/plugin/federated/engine_federated.cc +++ b/plugin/federated/engine_federated.cc @@ -84,17 +84,14 @@ class FederatedEngine : public IEngine { } } - int LoadCheckPoint(Serializable *global_model, Serializable *local_model = nullptr) override { + int LoadCheckPoint() override { return 0; } - void CheckPoint(const Serializable *global_model, - const Serializable *local_model = nullptr) override { + void CheckPoint() override { version_number_ += 1; } - void LazyCheckPoint(const Serializable *global_model) override { version_number_ += 1; } - int VersionNumber() const override { return version_number_; } /*! \brief get rank of current node */