From c4db8e408e4499f473208d7b0ee94b4a0eb0aa11 Mon Sep 17 00:00:00 2001 From: Arunima Chaudhuri Date: Wed, 27 Mar 2024 11:22:56 +0000 Subject: [PATCH] store config shape in pickle files Signed-off-by: Arunima Chaudhuri --- DAS/simulator.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/DAS/simulator.py b/DAS/simulator.py index a932e97..8a1867f 100644 --- a/DAS/simulator.py +++ b/DAS/simulator.py @@ -277,6 +277,13 @@ class Simulator: malicious_nodes_not_added_count = 0 steps = 0 unique_run_id = str(uuid.uuid4()) + backup_folder = f"results/{self.execID}/backup" + if not os.path.exists(backup_folder): + os.makedirs(backup_folder) + backup_file = os.path.join(backup_folder, f"simulation_data_{unique_run_id}.pkl") + + with open(backup_file, 'ab') as f: + pickle.dump(self.shape.__dict__, f) while(True): vectors_data = [] missingVector.append(missingSamples) @@ -374,7 +381,7 @@ class Simulator: 'columnNeighbors': list(self.validators[i].columnNeighbors) } vectors_data.append(validator_data) - + # Alse store for initNetwork vectors_data += (progressVector,missingVector) backup_folder = f"results/{self.execID}/backup" if not os.path.exists(backup_folder):