explicit casting for int overflow logic (#49)

sixlettervariables · Oct 25, 2024 · 4a156cc · 4a156cc
1 parent 93d86af
commit 4a156cc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pysierraecg/src/sierraecg/xli.py b/pysierraecg/src/sierraecg/xli.py
@@ -52,5 +52,6 @@ def xli_unpack(buffer: List[int]) -> npt.NDArray[np.int16]:
         [0 for _ in range(int(len(buffer) / 2))], dtype=np.int16
     )
     for i in range(len(unpacked)):
-        unpacked[i] = (((buffer[i] << 8) | buffer[len(unpacked) + i]) << 16) >> 16
+        joined_bytes = (((buffer[i] << 8) | buffer[len(unpacked) + i]) << 16) >> 16
+        unpacked[i] = np.array(joined_bytes).astype(np.int16)
     return unpacked