relax slot comparisons

drift-labs · Dec 5, 2023 · 4021da7 · 4021da7
1 parent 2cae1c6
commit 4021da7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/driftpy/accounts/bulk_account_loader.py b/src/driftpy/accounts/bulk_account_loader.py
@@ -147,7 +147,7 @@ async def load_chunk(self, chunk: List[List[AccountToLoad]]):
                 pubkey_str = str(account_to_load.pubkey)
                 old_buffer_and_slot = self.buffer_and_slot_map.get(pubkey_str)
 
-                if old_buffer_and_slot is not None and slot <= old_buffer_and_slot.slot:
+                if old_buffer_and_slot is not None and slot < old_buffer_and_slot.slot:
                     continue
 
                 new_buffer = None

diff --git a/src/driftpy/accounts/ws/account_subscriber.py b/src/driftpy/accounts/ws/account_subscriber.py
@@ -86,7 +86,7 @@ def _update_data(self, new_data: Optional[DataAndSlot[T]]):
         if new_data is None:
             return
 
-        if self.data_and_slot is None or new_data.slot > self.data_and_slot.slot:
+        if self.data_and_slot is None or new_data.slot >= self.data_and_slot.slot:
             self.data_and_slot = new_data
 
     def unsubscribe(self):