Updated a regext to be more secure.

XAITK · Dec 13, 2024 · 57e094a · 57e094a
1 parent 8441df4
commit 57e094a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -12,7 +12,7 @@ def parse_snapshot_to_numpy_no_eval(self, snapshot: str) -> tuple[np.ndarray]:
         snapshot = "\n".join(line for line in snapshot.splitlines() if not line.strip().startswith("#"))
 
         # Extract array strings using regex
-        array_pattern = r"array\((\[.*?\])\)"
+        array_pattern = r"array\((\[[^\]]*\])\)"
         matches = re.findall(array_pattern, snapshot, flags=re.S)
 
         # Parse each array string into a NumPy array