From 29d19f2cedc99ef0d923d40167686ea639654440 Mon Sep 17 00:00:00 2001 From: mohammadnaseri Date: Tue, 10 Sep 2024 22:25:27 +0100 Subject: [PATCH] Fix --- src/py/flwr/server/superlink/state/sqlite_state.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/py/flwr/server/superlink/state/sqlite_state.py b/src/py/flwr/server/superlink/state/sqlite_state.py index 199148b3045b..1c112285f489 100644 --- a/src/py/flwr/server/superlink/state/sqlite_state.py +++ b/src/py/flwr/server/superlink/state/sqlite_state.py @@ -632,7 +632,7 @@ def get_nodes(self, run_id: int) -> Set[int]: sint64_run_id = uint64_to_sint64(run_id) # Validate run ID - query = "SELECT COUNT(*) FROM run WHERE sint64_run_id = ?;" + query = "SELECT COUNT(*) FROM run WHERE run_id = ?;" if self.query(query, (sint64_run_id,))[0]["COUNT(*)"] == 0: return set() @@ -673,12 +673,12 @@ def create_run( sint64_run_id = uint64_to_sint64(run_id) # Check conflicts - query = "SELECT COUNT(*) FROM run WHERE sint64_run_id = ?;" + query = "SELECT COUNT(*) FROM run WHERE run_id = ?;" # If sint64_run_id does not exist if self.query(query, (sint64_run_id,))[0]["COUNT(*)"] == 0: query = ( "INSERT INTO run " - "(sint64_run_id, fab_id, fab_version, fab_hash, override_config)" + "(run_id, fab_id, fab_version, fab_hash, override_config)" "VALUES (?, ?, ?, ?, ?);" ) if fab_hash: