Fix error

adap · Apr 4, 2024 · 100eadb · 100eadb
1 parent 28afce5
commit 100eadb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/py/flwr/server/superlink/state/state.py b/src/py/flwr/server/superlink/state/state.py
@@ -179,6 +179,7 @@ def store_client_public_key(self, public_key: bytes) -> None:
     def get_client_public_keys(self) -> Set[bytes]:
         """Retrieve all currently stored `client_public_keys` as a set."""
 
+    @abc.abstractmethod
     def acknowledge_ping(self, node_id: int, ping_interval: float) -> bool:
         """Acknowledge a ping received from a node, serving as a heartbeat.
 

diff --git a/src/py/flwr/server/superlink/state/state_test.py b/src/py/flwr/server/superlink/state/state_test.py
@@ -555,7 +555,7 @@ def test_initialize(self) -> None:
         result = state.query("SELECT name FROM sqlite_schema;")
 
         # Assert
-        assert len(result) == 12
+        assert len(result) == 13
 
 
 class SqliteFileBasedTest(StateTest, unittest.TestCase):
@@ -580,7 +580,7 @@ def test_initialize(self) -> None:
         result = state.query("SELECT name FROM sqlite_schema;")
 
         # Assert
-        assert len(result) == 12
+        assert len(result) == 13
 
 
 if __name__ == "__main__":