NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-25 17:10:42 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

chore: refine code documentation and remove redundant annotations

Browse source
This commit is contained in:
RUFFY-369 2026-04-05 03:13:24 +05:30
parent 9955b245f8
commit 210883f0da
5 changed files with 12 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

12
atroposlib/api/shm_buffer.py
View file

@ -108,14 +108,14 @@ class ZeroCopySHMBuffer:
# Calculate offset in data segment
offset = SHMBufferConfig.SIZE + (write_idx * self.slot_size)
# 1. Write Score (float64, 8 bytes)
# Write Score
struct.pack_into("d", self.buf, offset, float(score))
# 2. Write Token Length (int32, 4 bytes)
# Write Token Length
token_len = min(len(tokens), entry_size)
struct.pack_into("i", self.buf, offset + 8, token_len)
# 3. Write Tokens (int32 array, EntrySize * 4 bytes)
# Write Tokens (Numpy View)
token_offset = offset + 8 + 4
token_arr = np.array(tokens, dtype=np.int32)
@ -140,14 +140,12 @@ class ZeroCopySHMBuffer:
offset = SHMBufferConfig.SIZE + (read_idx * self.slot_size)
# 1. Read Score (float64)
# Read Score and Token Length
score = struct.unpack_from("d", self.buf, offset)[0]
# 2. Read Token Length (int32)
token_len = struct.unpack_from("i", self.buf, offset + 8)[0]
token_len = min(token_len, entry_size)
# 3. Read Tokens (Numpy View)
# Read Tokens (Numpy View)
token_offset = offset + 8 + 4
tokens_view = np.ndarray((token_len,), dtype=np.int32, buffer=self.buf, offset=token_offset)