store config shape in pickle files

Signed-off-by: Arunima Chaudhuri <arunimachaudhuri2020@gmail.com>
This commit is contained in:
Arunima Chaudhuri 2024-03-27 11:22:56 +00:00
parent ec0f9cc781
commit c4db8e408e
1 changed files with 8 additions and 1 deletions

View File

@ -277,6 +277,13 @@ class Simulator:
malicious_nodes_not_added_count = 0 malicious_nodes_not_added_count = 0
steps = 0 steps = 0
unique_run_id = str(uuid.uuid4()) 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): while(True):
vectors_data = [] vectors_data = []
missingVector.append(missingSamples) missingVector.append(missingSamples)
@ -374,7 +381,7 @@ class Simulator:
'columnNeighbors': list(self.validators[i].columnNeighbors) 'columnNeighbors': list(self.validators[i].columnNeighbors)
} }
vectors_data.append(validator_data) vectors_data.append(validator_data)
# Alse store for initNetwork
vectors_data += (progressVector,missingVector) vectors_data += (progressVector,missingVector)
backup_folder = f"results/{self.execID}/backup" backup_folder = f"results/{self.execID}/backup"
if not os.path.exists(backup_folder): if not os.path.exists(backup_folder):