From 156df95fbf52a310650aad47074228cd04fe2b9f Mon Sep 17 00:00:00 2001
From: Gamelin Alexis <alexis.gamelin@synchrotron-soleil.fr>
Date: Fri, 20 Jan 2023 16:12:06 +0100
Subject: [PATCH] Update ImpedanceModel.load and save
---
mbtrack2/impedance/impedance_model.py | 9 +++++----
1 file changed, 5 insertions(+), 4 deletions(-)
diff --git a/mbtrack2/impedance/impedance_model.py b/mbtrack2/impedance/impedance_model.py
index e23b17a..4fc1352 100644
--- a/mbtrack2/impedance/impedance_model.py
+++ b/mbtrack2/impedance/impedance_model.py
@@ -507,7 +507,8 @@ class ImpedanceModel():
"""
to_save = {"wakefields":self.wakefields,
- "positions":self.positions}
+ "positions":self.positions,
+ "names":self.names}
with open(file,"wb") as f:
pickle.dump(to_save, f)
@@ -533,6 +534,6 @@ class ImpedanceModel():
to_load = pickle.load(f)
self.wakefields = to_load["wakefields"]
- self.positions = to_load["positions"]
- self.sum_elements()
- self.sum_by_name_all()
+ self.positions = to_load["positions"]
+ self.names = to_load["names"]
+ self.compute_sum()
--
GitLab