Fix network serialisability by filtering out Progress objects

This commit is contained in:
12v
2023-09-08 14:04:53 +01:00
parent b888b23595
commit 4115f9b5c3

View File

@@ -212,7 +212,7 @@ class Network:
def save(self, filename, location='../assets/networks/'):
"""Saves network in JSON format to '../assets/networks/'."""
network_data = {k: v for k, v in self.__dict__.items() if k != 'hop' and k != 'file'}
network_data = {k: v for k, v in self.__dict__.items() if k not in ['hop', 'file', 'progress']}
saved_network = json.dumps(network_data)
f = open(location + f'{filename}', 'w')
f.write(saved_network)