emu-base 2.2.1__py3-none-any.whl → 2.3.0__py3-none-any.whl

emu_base/__init__.py

@@ -5,6 +5,7 @@ from .math.krylov_exp import krylov_exp, DEFAULT_MAX_KRYLOV_DIM
 from .jump_lindblad_operators import compute_noise_from_lindbladians
 from .math.matmul import matmul_2x2_with_batched
 from .aggregators import AggregationType, aggregate
+from .utils import apply_measurement_errors
 
 __all__ = [
     "__version__",
@@ -18,6 +19,7 @@ __all__ = [
     "HamiltonianType",
     "DEFAULT_MAX_KRYLOV_DIM",
     "DEVICE_COUNT",
+    "apply_measurement_errors",
 ]
 
-__version__ = "2.2.1"
+__version__ = "2.3.0"

emu_base/noise.py

@@ -0,0 +1,11 @@
+import torch
+
+
+def pick_dark_qubits(eta: float, n: int) -> torch.Tensor:
+    """
+    Randomly pick n booleans such that ℙ(True) = eta
+
+    Returns:
+        A BoolTensor of shape (n,) where each element is True with probability eta.
+    """
+    return torch.rand(n) < eta

emu_base/pulser_adapter.py

@@ -294,6 +294,26 @@ def _get_all_lindblad_noise_operators(
     ]
 
 
+def _get_target_times(
+    sequence: pulser.Sequence, config: EmulationConfig, dt: int
+) -> list[int]:
+    sequence_duration = sequence.get_duration(include_fall_time=config.with_modulation)
+    # the end value is exclusive, so add +1
+    observable_times = set(range(0, sequence_duration + 1, dt))
+    observable_times.add(sequence_duration)
+    for obs in config.observables:
+        times: Sequence[float]
+        if obs.evaluation_times is not None:
+            times = obs.evaluation_times
+        elif config.default_evaluation_times != "Full":
+            times = config.default_evaluation_times.tolist()  # type: ignore[union-attr,assignment]
+        observable_times |= set([round(time * sequence_duration) for time in times])
+
+    target_times: list[int] = list(observable_times)
+    target_times.sort()
+    return target_times
+
+
 class PulserData:
     slm_end_time: float
     full_interaction_matrix: torch.Tensor
@@ -308,22 +328,7 @@ class PulserData:
     def __init__(self, *, sequence: pulser.Sequence, config: EmulationConfig, dt: int):
         self.qubit_ids = sequence.register.qubit_ids
         self.qubit_count = len(self.qubit_ids)
-        sequence_duration = sequence.get_duration()
-        # the end value is exclusive, so add +1
-        observable_times = set(torch.arange(0, sequence.get_duration() + 1, dt).tolist())
-        observable_times.add(sequence.get_duration())
-        for obs in config.observables:
-            times: Sequence[float]
-            if obs.evaluation_times is not None:
-                times = obs.evaluation_times
-            elif config.default_evaluation_times != "Full":
-                times = (
-                    config.default_evaluation_times.tolist()  # type: ignore[union-attr,assignment]
-                )
-            observable_times |= set([round(time * sequence_duration) for time in times])
-
-        self.target_times: list[int] = list(observable_times)
-        self.target_times.sort()
+        self.target_times = _get_target_times(sequence=sequence, config=config, dt=dt)
 
         laser_waist = config.noise_model.laser_waist
         amp_sigma = config.noise_model.amp_sigma

emu_base/utils.py

@@ -1,3 +1,5 @@
+from collections import Counter
+import random
 import torch
 
 
@@ -33,3 +35,39 @@ def deallocate_tensor(t: torch.Tensor) -> None:
     if t._base is not None:
         t._base.resize_(0)
         t._base.set_(source=replacement_storage)
+
+
+def readout_with_error(c: str, *, p_false_pos: float, p_false_neg: float) -> str:
+    # p_false_pos = false positive, p_false_neg = false negative
+    r = random.random()
+    if c == "0" and r < p_false_pos:
+        return "1"
+
+    if c == "1" and r < p_false_neg:
+        return "0"
+
+    return c
+
+
+def apply_measurement_errors(
+    bitstrings: Counter[str], *, p_false_pos: float, p_false_neg: float
+) -> Counter[str]:
+    """
+    Given a bag of sampled bitstrings, returns another bag of bitstrings
+    sampled with readout/measurement errors.
+
+        p_false_pos: probability of false positive
+        p_false_neg: probability of false negative
+    """
+
+    result: Counter[str] = Counter()
+    for bitstring, count in bitstrings.items():
+        for _ in range(count):
+            bitstring_with_error = "".join(
+                readout_with_error(c, p_false_pos=p_false_pos, p_false_neg=p_false_neg)
+                for c in bitstring
+            )
+
+            result[bitstring_with_error] += 1
+
+    return result

{emu_base-2.2.1.dist-info → emu_base-2.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: emu-base
-Version: 2.2.1
+Version: 2.3.0
 Summary: Pasqal base classes for emulators
 Project-URL: Documentation, https://pasqal-io.github.io/emulators/
 Project-URL: Repository, https://github.com/pasqal-io/emulators

{emu_base-2.2.1.dist-info → emu_base-2.3.0.dist-info}/RECORD

@@ -1,15 +1,16 @@
-emu_base/__init__.py,sha256=lXgYx3VbA0qNMwDZyV4HpFcr6aUDxhqWjy8Um73JXrs,681
+emu_base/__init__.py,sha256=wD_KJ7LfcWPbEbnskS46DOhToP4r6D_y2EyztecGMqg,757
 emu_base/aggregators.py,sha256=bB-rldoDAErxQMpL715K5lpiabGOpkCY0GyxW7mfHuc,5000
 emu_base/constants.py,sha256=41LYkKLUCz-oxPbd-j7nUDZuhIbUrnez6prT0uR0jcE,56
 emu_base/jump_lindblad_operators.py,sha256=Y30f8emVFS4Dazljc_Rh4lX9qU4QQY_AxPNahnzcsfY,2101
-emu_base/pulser_adapter.py,sha256=WJX8y6iBYg8FX--q0Y10haz-35TbmUGoHbMDQEDuZIY,14534
-emu_base/utils.py,sha256=-nIoJuu1pOxkPc2tiJTLjQ0ONsPR43CCB6vOJqdWaUc,1425
+emu_base/noise.py,sha256=h47DhEsFdAN783VnisJ50QJIiAPeRPKpU0SYtt8bAEQ,273
+emu_base/pulser_adapter.py,sha256=NlZ1nNuNdU7D8S0XJqtjgu1b-qW2jFVzXOCAroH4Vuk,14660
+emu_base/utils.py,sha256=17gca5uQfPaXiH3z4Leqd_rAJEZKQUqlSHj7OzFvCkU,2495
 emu_base/math/__init__.py,sha256=6BbIytYV5uC-e5jLMtIErkcUl_PvfSNnhmVFY9Il8uQ,97
 emu_base/math/brents_root_finding.py,sha256=AVx6L1Il6rpPJWrLJ7cn6oNmJyZOPRgEaaZaubC9lsU,3711
 emu_base/math/double_krylov.py,sha256=X16dyCbyzdP7fFK-hmKS03Q-DJtC6TZ8sJrGTJ6akIc,3708
 emu_base/math/krylov_energy_min.py,sha256=hm_B5qtBXHY1hl-r_LgDUKNDsdqVCDBHprQB3D-UFR8,4009
 emu_base/math/krylov_exp.py,sha256=mGFddVQ8mEbwypbZtnlRPFpi4Nf8JZT6OKLHloIwCDQ,3934
 emu_base/math/matmul.py,sha256=lEAnV0b5z_f1xEA-9p-WXxA8bM3QbShiHdXQ3ZkZFcQ,877
-emu_base-2.2.1.dist-info/METADATA,sha256=Rs1xk1Xu4iB3-sWXgq0ASiM2bULwk0DVEfC-AclHIGs,3604
-emu_base-2.2.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-emu_base-2.2.1.dist-info/RECORD,,
+emu_base-2.3.0.dist-info/METADATA,sha256=AAanXBVpBx2__3ojtFYDqPjgLF-9WqNmKcXH4V3JiJA,3604
+emu_base-2.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+emu_base-2.3.0.dist-info/RECORD,,