Skip to content

Commit

Permalink
Use JSON for sqlite state
Browse files Browse the repository at this point in the history
  • Loading branch information
charlesbvll committed Jul 8, 2024
1 parent 0b1da02 commit e45ffea
Showing 1 changed file with 5 additions and 3 deletions.
8 changes: 5 additions & 3 deletions src/py/flwr/server/superlink/state/sqlite_state.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,10 +15,10 @@
"""SQLite based implemenation of server state."""


import json
import re
import sqlite3
import time
from ast import literal_eval
from logging import DEBUG, ERROR
from typing import Any, Dict, List, Optional, Sequence, Set, Tuple, Union, cast
from uuid import UUID, uuid4
Expand Down Expand Up @@ -633,7 +633,9 @@ def create_run(
"INSERT INTO run (run_id, fab_id, fab_version, override_config)"
"VALUES (?, ?, ?, ?);"
)
self.query(query, (run_id, fab_id, fab_version, str(override_config)))
self.query(
query, (run_id, fab_id, fab_version, json.dumps(override_config))
)
return run_id
log(ERROR, "Unexpected run creation failure.")
return 0
Expand Down Expand Up @@ -700,7 +702,7 @@ def get_run(self, run_id: int) -> Optional[Run]:
run_id=run_id,
fab_id=row["fab_id"],
fab_version=row["fab_version"],
override_config=literal_eval(row["override_config"]),
override_config=json.loads(row["override_config"]),
)
except sqlite3.IntegrityError:
log(ERROR, "`run_id` does not exist.")
Expand Down

0 comments on commit e45ffea

Please sign in to comment.