From 0725fd60819f9758fbed6ee54f34f3696a2fb2f8 Mon Sep 17 00:00:00 2001
From: Rong Ou <rong.ou@gmail.com>
Date: Thu, 23 Jun 2022 17:41:07 -0700
Subject: [PATCH] fix federated learning plugin (#8027)

---
 plugin/federated/engine_federated.cc | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/plugin/federated/engine_federated.cc b/plugin/federated/engine_federated.cc
index 8fd396d94..9b43c3997 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 */