Skip to content

Commit 15d5ab2

Browse files
committed
refactor: migrate
1 parent 0731525 commit 15d5ab2

File tree

1 file changed

+8
-12
lines changed

1 file changed

+8
-12
lines changed

Diff for: src/modules/csm/state.py

+8-12
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,7 @@ class State:
5858
_consensus_version: int = 1
5959

6060
def __init__(self) -> None:
61+
self.frames = []
6162
self.data = {}
6263
self._epochs_to_process = tuple()
6364
self._processed_epochs = set()
@@ -153,21 +154,16 @@ def migrate(
153154
)
154155
self.clear()
155156

156-
frames = self._calculate_frames(tuple(sequence(l_epoch, r_epoch)), epochs_per_frame)
157+
new_frames = self._calculate_frames(tuple(sequence(l_epoch, r_epoch)), epochs_per_frame)
158+
if self.frames == new_frames:
159+
logger.info({"msg": "No need to migrate duties data cache"})
160+
return
161+
self._migrate_frames_data(new_frames)
157162

158-
if not self.is_empty:
159-
cached_frames = self.frames
160-
if cached_frames == frames:
161-
logger.info({"msg": "No need to migrate duties data cache"})
162-
return
163-
self._migrate_frames_data(frames)
164-
else:
165-
self.data = {frame: defaultdict(AttestationsAccumulator) for frame in frames}
166-
167-
self.frames = frames
163+
self.frames = new_frames
164+
self.find_frame.cache_clear()
168165
self._epochs_to_process = tuple(sequence(l_epoch, r_epoch))
169166
self._consensus_version = consensus_version
170-
self.find_frame.cache_clear()
171167
self.commit()
172168

173169
def _migrate_frames_data(self, new_frames: list[Frame]):

0 commit comments

Comments
 (0)