emu-base 2.4.3__py3-none-any.whl → 2.5.0__py3-none-any.whl

emu_base/__init__.py

@@ -4,7 +4,7 @@ from .math.brents_root_finding import find_root_brents
 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 .utils import get_max_rss, apply_measurement_errors, unix_like
+from .utils import get_max_rss, apply_measurement_errors, unix_like, init_logging
 
 __all__ = [
     "__version__",
@@ -19,6 +19,7 @@ __all__ = [
     "DEVICE_COUNT",
     "apply_measurement_errors",
     "unix_like",
+    "init_logging",
 ]
 
-__version__ = "2.4.3"
+__version__ = "2.5.0"

emu_base/jump_lindblad_operators.py

@@ -6,7 +6,11 @@ dtype = torch.complex128
 
 
 def get_lindblad_operators(
-    *, noise_type: str, noise_model: NoiseModel, dim: int = 2
+    *,
+    noise_type: str,
+    noise_model: NoiseModel,
+    interact_type: str = "ising",
+    dim: int = 2,
 ) -> list[torch.Tensor]:
 
     assert noise_type in noise_model.noise_types
@@ -19,7 +23,9 @@ def get_lindblad_operators(
 
     if noise_type == "dephasing":
         if noise_model.hyperfine_dephasing_rate != 0.0:
-            raise NotImplementedError("hyperfine_dephasing_rate is unsupported")
+            raise NotImplementedError(
+                "hyperfine_dephasing_rate is supported only in the digital basis"
+            )
 
         c = math.sqrt(noise_model.dephasing_rate / 2)
         dephasing = torch.zeros(dim, dim, dtype=dtype)
@@ -51,15 +57,24 @@ def get_lindblad_operators(
             for op in noise_model.eff_noise_opers
         ):
             raise ValueError(
-                "Only 2 by 2 or 3 by 3 effective noise operator matrices are supported"
+                f"Only {dim} by {dim} effective noise operator matrices are "
+                "supported and it should be given as torch tensors "
             )
 
-        return [  # lindblad operators are coming from pulser
-            math.sqrt(rate)
-            * torch.flip(op if isinstance(op, torch.Tensor) else torch.tensor(op), (0, 1))
+        lindblad_ops = [  # lindblad operators with XY pulser basis are fine
+            math.sqrt(rate) * torch.as_tensor(op)
             for rate, op in zip(noise_model.eff_noise_rates, noise_model.eff_noise_opers)
         ]
 
+        # pulser ising basis changing to emu-mps ising basis
+        if interact_type == "ising":
+            for tensor in lindblad_ops:
+                tensor[:2, :2] = torch.flip(tensor[:2, :2], (0, 1))
+
+        return lindblad_ops
+    if noise_type == "leakage":  # leakage operators are eff_noise
+        return []
+
     raise ValueError(f"Unknown noise type: {noise_type}")
 
 

emu_base/pulser_adapter.py

@@ -20,7 +20,7 @@ _NON_LINDBLADIAN_NOISE = {"SPAM", "doppler", "amplitude", "detuning", "register"
 
 
 def _get_all_lindblad_noise_operators(
-    noise_model: NoiseModel | None, dim: int = 2
+    noise_model: NoiseModel | None, dim: int = 2, interact_type: str = "ising"
 ) -> list[torch.Tensor]:
     if noise_model is None:
         return []
@@ -30,7 +30,10 @@ def _get_all_lindblad_noise_operators(
         for noise_type in noise_model.noise_types
         if noise_type not in _NON_LINDBLADIAN_NOISE
         for op in get_lindblad_operators(
-            noise_type=noise_type, noise_model=noise_model, dim=dim
+            noise_type=noise_type,
+            noise_model=noise_model,
+            dim=dim,
+            interact_type=interact_type,
         )
     ]
 
@@ -150,6 +153,7 @@ class PulserData:
         self.omega, self.delta, self.phi = _extract_omega_delta_phi(
             self.hamiltonian.noisy_samples, self.qubit_ids, self.target_times
         )
+        self.eigenstates = self.hamiltonian.eigenbasis
 
         int_type = self.hamiltonian.interaction_type
         self.dim = self.hamiltonian.dim
@@ -161,7 +165,7 @@ class PulserData:
             raise ValueError(f"Unsupported basis: {int_type}")
 
         self.lindblad_ops = _get_all_lindblad_noise_operators(
-            config.noise_model, dim=self.dim
+            config.noise_model, dim=self.dim, interact_type=int_type
         )
         self.has_lindblad_noise: bool = self.lindblad_ops != []
 

emu_base/utils.py

@@ -2,12 +2,33 @@ from collections import Counter
 import random
 import torch
 import os
+import logging
+import sys
+from pathlib import Path
 
 unix_like = os.name != "nt"
 if unix_like:
     from resource import RUSAGE_SELF, getrusage
 
 
+def init_logging(log_level: int, log_file: Path | None) -> logging.Logger:
+    logger = logging.getLogger("emulators")
+    logger.propagate = False
+
+    handler: logging.Handler
+    if log_file is None:
+        handler = logging.StreamHandler(sys.stdout)
+    else:
+        handler = logging.FileHandler(log_file, mode="w")
+
+    handler.setLevel(log_level)
+    handler.setFormatter(logging.Formatter("%(message)s"))
+    for h in logger.handlers:
+        logger.removeHandler(h)
+    logger.addHandler(handler)
+    return logger
+
+
 def deallocate_tensor(t: torch.Tensor) -> None:
     """
     Free the memory used by a tensor. This is done regardless of the

{emu_base-2.4.3.dist-info → emu_base-2.5.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: emu-base
-Version: 2.4.3
+Version: 2.5.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.4.3.dist-info → emu_base-2.5.0.dist-info}/RECORD

@@ -1,14 +1,14 @@
-emu_base/__init__.py,sha256=2_F4En9Wj4MwnJXavLN7LqV5CnOnV0JGSCFBSCuGhsw,725
+emu_base/__init__.py,sha256=0gT5wd46vY1MykuZ-E1-h9Ev1fOXE_tu59FzRDH9PCM,759
 emu_base/constants.py,sha256=41LYkKLUCz-oxPbd-j7nUDZuhIbUrnez6prT0uR0jcE,56
-emu_base/jump_lindblad_operators.py,sha256=b5xL62aCVFjbNRigyBtRTacG4Xg2A3eYXRe2KBNlt9w,2588
-emu_base/pulser_adapter.py,sha256=NB220g-ESyi8Te-mcmj0ciLpi4JG-Bj7dVKy5sKJuUU,7099
-emu_base/utils.py,sha256=OoJ0GjL1K8m9Jq-BeBfmbWuFp9bvwXp8UcI3BMW94Js,2974
+emu_base/jump_lindblad_operators.py,sha256=F_C-1aruXY_SLzw0m_HD5loqRGdjgsLHEzttlMzWNEw,3015
+emu_base/pulser_adapter.py,sha256=o30fm2soOXr7y2hvoZKLC24oi-FW0_6RmSgrsSWu_y4,7274
+emu_base/utils.py,sha256=kXfRFq31r_dfLbBC0SHiPpvf7Wll4behd2vqQJsTrVo,3553
 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=iR4hmE0eXptbAg3opikd5d4Zv7dhnDrawH-n_4KG-cc,4009
 emu_base/math/krylov_exp.py,sha256=mGFddVQ8mEbwypbZtnlRPFpi4Nf8JZT6OKLHloIwCDQ,3934
 emu_base/math/matmul.py,sha256=lEAnV0b5z_f1xEA-9p-WXxA8bM3QbShiHdXQ3ZkZFcQ,877
-emu_base-2.4.3.dist-info/METADATA,sha256=atbuyLSrlYZNsVpLubWaBV4yblkgllwW45DvmFKvZlE,3604
-emu_base-2.4.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-emu_base-2.4.3.dist-info/RECORD,,
+emu_base-2.5.0.dist-info/METADATA,sha256=U1_pInFyZ7gjemxHzZjX1F4y3hfhkw174NjuPVw47gI,3604
+emu_base-2.5.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+emu_base-2.5.0.dist-info/RECORD,,