Saving and loading from json

MaastrichtU-CDS · Aug 28, 2023 · 49cb712 · 49cb712
1 parent 2d63c66
commit 49cb712
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/helper_server.py b/helper_server.py
@@ -40,7 +40,7 @@ def initialize(model_centroids: str, **kwargs) -> Dict[str, Any]:
     # Save initial centroids
     print("Saving initial centroids")
     with open(model_centroids, "w+") as f:
-        json.dump(kwargs["centroids"], f)
+        json.dump({"centroids": kwargs["centroids"]}, f)
 
     return kwargs
 
@@ -80,8 +80,8 @@ def average_weights(model_centroids: str, full_model: str, files: List[str], k:
     # Save results
     print("Saving average centroids")
     with open(model_centroids, "w+") as f:
-        json.dump(centroids, f)
+        json.dump({"centroids": centroids}, f)
     with open(full_model, "w+") as f:
-        json.dump(centroids, f)
+        json.dump({"centroids": centroids}, f)
 
     return change
diff --git a/helper_worker.py b/helper_worker.py
@@ -5,7 +5,7 @@
 import numpy as np
 from pandas.api.types import is_string_dtype
 from scipy.spatial import distance
-from typing import Optional, Any, Dict, Callable, Tuple, List
+from typing import Optional, List
 
 logger = logging.getLogger("helper_worker_logger")
 data: Optional[np.ndarray] = None
@@ -69,8 +69,7 @@ def train(key: str, starting_centroids: str, k: int, columns: list):
         for column in columns:
             centroid.append(dfc[column].mean())
         centroids.append(centroid)
-    centroids = {"centroids": centroids}
 
     print("Saving local centroids")
     with open(f"centroids.{key}.json", "w+") as f:
-        json.dump(centroids, f)
+        json.dump({"centroids": centroids}, f)