emu-sv 2.4.0__py3-none-any.whl → 2.4.1__py3-none-any.whl

emu_sv/__init__.py

@@ -38,4 +38,4 @@ __all__ = [
     "DensityMatrix",
 ]
 
-__version__ = "2.4.0"
+__version__ = "2.4.1"

emu_sv/state_vector.py

@@ -275,7 +275,7 @@ class StateVector(State[complex, torch.Tensor]):
         return accum_state, amplitudes
 
     def overlap(self, other: StateVector, /) -> torch.Tensor:
-        return self.inner(other)
+        return torch.abs(self.inner(other)) ** 2
 
 
 def inner(left: StateVector, right: StateVector) -> torch.Tensor:

emu_sv/sv_config.py

@@ -6,6 +6,8 @@ from types import MethodType
 from typing import Any, ClassVar
 
 from emu_sv.utils import choose
+from emu_sv.state_vector import StateVector
+from emu_sv.dense_operator import DenseOperator
 
 from emu_sv.custom_callback_implementations import (
     qubit_occupation_sv_impl,
@@ -59,6 +61,8 @@ class SVConfig(EmulationConfig):
 
     # Whether to warn if unexpected kwargs are received
     _enforce_expected_kwargs: ClassVar[bool] = True
+    _state_type = StateVector
+    _operator_type = DenseOperator
 
     def __init__(
         self,

emu_sv/time_evolution.py

@@ -140,7 +140,10 @@ class EvolveStateVector(torch.autograd.Function):
             interaction_matrix=interaction_matrix,
             device=state.device,
         )
-        op = lambda x: -1j * dt * (ham * x)
+
+        def op(x: torch.Tensor) -> torch.Tensor:
+            return -1j * dt * (ham * x)
+
         res = krylov_exp(
             op,
             state,
@@ -273,7 +276,10 @@ class EvolveStateVector(torch.autograd.Function):
         )
 
         if any(ctx.needs_input_grad[1:5]):
-            op = lambda x: -1j * dt * (ham * x)
+
+            def op(x: torch.Tensor):
+                return -1j * dt * (ham * x)
+
             lanczos_vectors_state, dS, lanczos_vectors_grad = double_krylov(
                 op, state, grad_state_out, tolerance
             )
@@ -316,7 +322,10 @@ class EvolveStateVector(torch.autograd.Function):
                     grad_int_mat[i, j] = (-1j * dt * torch.tensordot(Vg.conj(), v)).real
 
         if ctx.needs_input_grad[5]:
-            op = lambda x: (1j * dt) * (ham * x)
+
+            def op(x: torch.Tensor):
+                return (1j * dt) * (ham * x)
+
             grad_state_in = krylov_exp(op, grad_state_out.detach(), tolerance, tolerance)
 
         return (
@@ -353,7 +362,9 @@ class EvolveDensityMatrix:
             device=density_matrix.device,
         )
 
-        op = lambda x: -1j * dt * (ham @ x)
+        def op(x: torch.Tensor) -> torch.Tensor:
+            return -1j * dt * (ham @ x)
+
         return (
             krylov_exp(
                 op,

{emu_sv-2.4.0.dist-info → emu_sv-2.4.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: emu-sv
-Version: 2.4.0
+Version: 2.4.1
 Summary: Pasqal State Vector based pulse emulator built on PyTorch
 Project-URL: Documentation, https://pasqal-io.github.io/emulators/
 Project-URL: Repository, https://github.com/pasqal-io/emulators
@@ -25,7 +25,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy
 Requires-Python: >=3.10
-Requires-Dist: emu-base==2.4.0
+Requires-Dist: emu-base==2.4.1
 Description-Content-Type: text/markdown
 
 <div align="center">

{emu_sv-2.4.0.dist-info → emu_sv-2.4.1.dist-info}/RECORD

@@ -1,15 +1,15 @@
-emu_sv/__init__.py,sha256=Oc9_UlpE7Nioggmg-pX4_8iQnh7_o9jl8n_wBTu5NJg,771
+emu_sv/__init__.py,sha256=Sv7BbiGYuZzUh30AqMZpyA9m2zM8z-UKh_D2C6LACNs,771
 emu_sv/custom_callback_implementations.py,sha256=_7XLIDzJ-p3DVqz-Jyv0eYbl8nih2x2p-pM4cBCLumA,6367
 emu_sv/dense_operator.py,sha256=AvgntJNwwtf3Wl66CIWSwUezVYV3vignCL24SQQQwQg,5901
 emu_sv/density_matrix_state.py,sha256=6QmLZvqEHLR64r0nD7D2jZIiAYOgciNVCjh3ywfvIs0,7243
 emu_sv/hamiltonian.py,sha256=CqNGuWJlO2ZljK47wt130s-5uKiOldQUsC3tjwk1mKA,6106
 emu_sv/lindblad_operator.py,sha256=pgjRNLBcvEM2-qxM8uy9wL74OtrD4A8trQeERi_AXH8,8892
-emu_sv/state_vector.py,sha256=dkAnxEQxK-4Fn6UxhDbcb1z1EsHFUpOyVy0mjHkKFBI,9894
+emu_sv/state_vector.py,sha256=Ookw_BDqzs9OFppQd7kaMxDSkaj5F_clhOCgo-LSaMg,9910
 emu_sv/sv_backend.py,sha256=-soOkSEzEBK1dCKnYnbtvYjmNZtZra1_4jP3H1ROOtM,737
 emu_sv/sv_backend_impl.py,sha256=OYsJK_DJ8dmq3QZLKiBejXrq22R5Auyye-pai07NeeY,8798
-emu_sv/sv_config.py,sha256=J4yb9-Na5hAbShDySzPfbt0WXvL8JpfrTEikTK8nH9s,5943
-emu_sv/time_evolution.py,sha256=_VH4f2RF6lGKzO08WxTYJ5Aw8_pTTMRKcyMnIuxH03I,13382
+emu_sv/sv_config.py,sha256=6fcuC9RKc7Pp55Yb-lWCGtZ7UJP53oLyZsy1f9wBtRg,6100
+emu_sv/time_evolution.py,sha256=Uy3qMdt3BlLB6Aq1-o5uajRTu_3fPuBCtcusHxFPPJc,13545
 emu_sv/utils.py,sha256=t0nMDVo6DF5bQW-vbsyRMCmvkyNxCU-v0Enmns9aOAU,1151
-emu_sv-2.4.0.dist-info/METADATA,sha256=HKTJmI_Ra3dcFw2QthE2TEvEiYcvcZmrHkrFMuGY794,3595
-emu_sv-2.4.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-emu_sv-2.4.0.dist-info/RECORD,,
+emu_sv-2.4.1.dist-info/METADATA,sha256=5nUzMJNramhABPKHG1D7h78W6L2MeSJpkZg1wAmDg2c,3595
+emu_sv-2.4.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+emu_sv-2.4.1.dist-info/RECORD,,