ket-lang 0.8.3.dev0__tar.gz → 0.8.4.dev0__tar.gz

{ket_lang-0.8.3.dev0/src/ket_lang.egg-info → ket_lang-0.8.4.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ket-lang
-Version: 0.8.3.dev0
+Version: 0.8.4.dev0
 Summary: Ket quantum programming language interpreter and library
 Home-page: https://quantumket.org
 Author: Evandro Chagas Ribeiro da Rosa

{ket_lang-0.8.3.dev0 → ket_lang-0.8.4.dev0}/src/ket/__init__.py

@@ -90,7 +90,7 @@ from .expv import __all__ as all_expv
 from .quantumstate import *
 from .quantumstate import __all__ as all_state
 
-__version__ = "0.8.3.dev0"
+__version__ = "0.8.4.dev0"
 
 __all__ = all_base + all_func + all_gate + all_expv + all_state
 

{ket_lang-0.8.3.dev0 → ket_lang-0.8.4.dev0}/src/ket/base.py

@@ -677,14 +677,6 @@ class Samples:
             **kwargs,
         )
 
-        fig.update_layout(
-            bargap=0.75,
-            xaxis={
-                "tickmode": "linear",
-                "dtick": 1,
-            },
-        )
-
         return fig
 
     def __repr__(self) -> str:

{ket_lang-0.8.3.dev0 → ket_lang-0.8.4.dev0}/src/ket/gates.py

@@ -26,8 +26,8 @@ Example:
         from math import pi
         from ket import *
 
-        s_gate = PHASE(pi/2)
-        t_gate = PHASE(pi/4)
+        s_gate = P(pi/2)
+        t_gate = P(pi/4)
 
         p = Process()
         q = p.alloc()
@@ -47,6 +47,7 @@ from math import pi
 from functools import reduce
 from operator import add
 from typing import Any, Callable
+import warnings
 
 from .clib.libket import (
     HADAMARD,
@@ -72,12 +73,14 @@ __all__ = [
     "RY",
     "RZ",
     "PHASE",
+    "P",
     "S",
     "T",
     "SD",
     "TD",
     "U3",
     "CNOT",
+    "CZ",
     "SWAP",
     "RXX",
     "RZZ",
@@ -93,7 +96,7 @@ def _gate_docstring(name, matrix, effect=None) -> str:
     
     .. csv-table::
         :delim: ;
-        :header: Matrix{", Effect" if effect is not None else ""}
+        :header: Matrix{"; Effect" if effect is not None else ""}
 
         :math:`{matrix}`{f"; :math:`{effect}`" if effect is not None else ""}
     """
@@ -277,7 +280,7 @@ RZ.__doc__ = _gate_docstring(
 )
 
 
-def PHASE(  # pylint: disable=invalid-name missing-function-docstring
+def P(  # pylint: disable=invalid-name missing-function-docstring
     theta: float, qubits: Quant | None = None
 ) -> Quant | Callable[[Quant], Quant]:
 
@@ -294,6 +297,17 @@ def PHASE(  # pylint: disable=invalid-name missing-function-docstring
     return inner(qubits)
 
 
+def PHASE(  # pylint: disable=invalid-name missing-function-docstring
+    theta: float, qubits: Quant | None = None
+) -> Quant | Callable[[Quant], Quant]:
+    warnings.warn(
+        "PHASE is deprecated and will be removed in future versions. Use P(theta, qubits) instead.",
+        DeprecationWarning,
+        stacklevel=2,
+    )
+    return P(theta, qubits)
+
+
 PHASE.__doc__ = _gate_docstring(
     "Phase shift",
     r"\begin{bmatrix} 1 & 0 \\ 0 & e^{i\theta} \end{bmatrix}",
@@ -302,7 +316,15 @@ PHASE.__doc__ = _gate_docstring(
 )
 
 
-S = PHASE(pi / 2)
+P.__doc__ = _gate_docstring(
+    "Phase shift",
+    r"\begin{bmatrix} 1 & 0 \\ 0 & e^{i\theta} \end{bmatrix}",
+    r"\begin{matrix} P\left|0\right> = & \left|0\right> \\"
+    r"P\left|1\right> = & e^{i\theta}\left|1\right> \end{matrix}",
+)
+
+
+S = P(pi / 2)
 S.__doc__ = _gate_docstring(
     "S",
     r"\begin{bmatrix} 1 & 0 \\ 0 & i \end{bmatrix}",
@@ -310,7 +332,7 @@ S.__doc__ = _gate_docstring(
     r"S\left|1\right> = & i\left|1\right> \end{matrix}",
 )
 
-SD = PHASE(-pi / 2)
+SD = P(-pi / 2)
 SD.__doc__ = _gate_docstring(
     "S-dagger",
     r"\begin{bmatrix} 1 & 0 \\ 0 & -i \end{bmatrix}",
@@ -318,7 +340,7 @@ SD.__doc__ = _gate_docstring(
     r"S^\dagger\left|1\right> = & -i\left|1\right> \end{matrix}",
 )
 
-T = PHASE(pi / 4)
+T = P(pi / 4)
 T.__doc__ = _gate_docstring(
     "T",
     r"\begin{bmatrix} 1 & 0 \\ 0 & e^{i\pi/4} \end{bmatrix}",
@@ -326,7 +348,7 @@ T.__doc__ = _gate_docstring(
     r"T\left|1\right> = & e^{i\pi/4}\left|1\right> \end{matrix}",
 )
 
-TD = PHASE(-pi / 4)
+TD = P(-pi / 4)
 TD.__doc__ = _gate_docstring(
     "T-dagger",
     r"\begin{bmatrix} 1 & 0 \\ 0 & e^{-i\pi/4} \end{bmatrix}",
@@ -359,6 +381,30 @@ CNOT.__doc__ = _gate_docstring(
 )
 
 
+def CZ(  # pylint: disable=invalid-name missing-function-docstring
+    *qubits: list[Quant],
+) -> list[Quant, Quant]:
+    for q in zip(*qubits):
+        ctrl(q[:-1], X)(q[-1])
+    return qubits
+
+
+CZ.__doc__ = _gate_docstring(
+    "Multi-Controlled Z",
+    r"\begin{bmatrix}"
+    r"1 & 0 & \cdots & 0 & 0 \\ 0 & 1 & \cdots & 0 & 0 \\"
+    r"\vdots & \vdots & \ddots & \vdots & \vdots \\ 0 & 0 & \cdots & 1 & 0 \\"
+    r"0 & 0 & \cdots & 0 & -1"
+    r"\end{bmatrix}",
+    r"\begin{matrix}"
+    r"\text{CZ}\left|0\cdots0\right> = & \left|0\cdots0\right> \\"
+    r"\text{CZ}\left|0\cdots1\right> = & \left|0\cdots1\right> \\"
+    r"\text{CZ}\left|1\cdots0\right> = & \left|1\cdots0\right> \\"
+    r"\text{CZ}\left|1\cdots1\right> = & -\left|1\cdots1\right> \\"
+    r"\end{matrix}",
+)
+
+
 def SWAP(  # pylint: disable=invalid-name missing-function-docstring
     qubit_a: Quant, qubit_b: Quant
 ) -> tuple[Quant, Quant]:
@@ -386,7 +432,7 @@ SWAP.__doc__ = _gate_docstring(
 
 
 def RXX(  # pylint: disable=invalid-name missing-function-docstring
-    theta: float, qubits_a: Quant | None, qubits_b: Quant | None
+    theta: float, qubits_a: Quant | None = None, qubits_b: Quant | None = None
 ) -> tuple[Quant, Quant] | Callable[[Quant, Quant], tuple[Quant, Quant]]:
     def inner(qubits_a: Quant, qubits_b: Quant) -> tuple[Quant, Quant]:
         for qubit_a, qubit_b in zip(qubits_a, qubits_b):
@@ -410,7 +456,7 @@ RXX.__doc__ = _gate_docstring(
 
 
 def RZZ(  # pylint: disable=invalid-name missing-function-docstring
-    theta: float, qubits_a: Quant | None, qubits_b: Quant | None
+    theta: float, qubits_a: Quant | None = None, qubits_b: Quant | None = None
 ) -> tuple[Quant, Quant] | Callable[[Quant, Quant], tuple[Quant, Quant]]:
     def inner(qubits_a: Quant, qubits_b: Quant) -> tuple[Quant, Quant]:
         for qubit_a, qubit_b in zip(qubits_a, qubits_b):
@@ -434,7 +480,7 @@ RZZ.__doc__ = _gate_docstring(
 
 
 def RYY(  # pylint: disable=invalid-name missing-function-docstring
-    theta: float, qubits_a: Quant | None, qubits_b: Quant | None
+    theta: float, qubits_a: Quant | None = None, qubits_b: Quant | None = None
 ) -> tuple[Quant, Quant] | Callable[[Quant, Quant], tuple[Quant, Quant]]:
     def inner(qubits_a: Quant, qubits_b: Quant) -> tuple[Quant, Quant]:
         for qubit_a, qubit_b in zip(qubits_a, qubits_b):
@@ -459,7 +505,7 @@ RYY.__doc__ = _gate_docstring(
 
 
 def RBS(  # pylint: disable=invalid-name missing-function-docstring
-    theta: float, qubits_a: Quant | None, qubits_b: Quant | None
+    theta: float, qubits_a: Quant | None = None, qubits_b: Quant | None = None
 ) -> tuple[Quant, Quant] | Callable[[Quant, Quant], tuple[Quant, Quant]]:
     def inner(qubits_a: Quant, qubits_b: Quant) -> tuple[Quant, Quant]:
         for qubit_a, qubit_b in zip(qubits_a, qubits_b):

{ket_lang-0.8.3.dev0 → ket_lang-0.8.4.dev0}/src/ket/lib/__init__.py

@@ -140,18 +140,17 @@ def dump_matrix(
     gate: Callable,
     num_qubits: int | list[int] = 1,
     args=(),
-    simulator: Literal["sparse", "dense"] = "sparse",
+    process: Process | None = None,
 ) -> list[list[complex]]:
     """Get the matrix representation of a quantum gate.
 
-    This function calculates the matrix representation of a quantum gate using
-    the specified simulator.
+    This function calculates the matrix representation of a quantum gate.
 
     Args:
         gate: Quantum gate operation to obtain the matrix for.
         num_qubits: Number of qubits.
         args: Classical arguments to pass to the gate function.
-        simulator: Simulator type to use.
+        process: Quantum process used to generate the matrix.
 
     Returns:
         Matrix representation of the quantum gate.
@@ -164,7 +163,12 @@ def dump_matrix(
 
     num_qubits = sum(num_qubits) if isinstance(num_qubits, Iterable) else num_qubits
 
-    process = Process(num_qubits=2 * num_qubits, execution="batch", simulator=simulator)
+    if process is None:
+        process = Process(
+            num_qubits=2 * num_qubits,
+            execution="batch",
+            simulator="sparse",
+        )
 
     mat = [[0.0j for _ in range(2**num_qubits)] for _ in range(2**num_qubits)]
 
@@ -325,6 +329,39 @@ def unitary(matrix: list[list[complex]]) -> Callable[[Quant], Quant]:
     return inner
 
 
+DRAW_STYLE = {
+    "textcolor": "#0f1f2e",
+    "gatetextcolor": "#f4f8ff",
+    "subtextcolor": "#f4f8ff",
+    "linecolor": "#0f1f2e",
+    "creglinecolor": "#6b7682",
+    "gatefacecolor": "#0f7cfb",
+    "barrierfacecolor": "#c6cdd5",
+    "backgroundcolor": "#f4f8ff",
+    "displaytext": {
+        "u1": "P",
+        "rx": "RX",
+        "ry": "RY",
+        "rz": "RZ",
+    },
+    "displaycolor": {
+        "u1": ["#008699", "#f4f8ff"],
+        "x": ["#0f7cfb", "#f4f8ff"],
+        "y": ["#0c63c9", "#f4f8ff"],
+        "z": ["#009cb3", "#f4f8ff"],
+        "h": ["#3f96fc", "#f4f8ff"],
+        "rx": ["#063264", "#f4f8ff"],
+        "ry": ["#094a97", "#f4f8ff"],
+        "rz": ["#008699", "#f4f8ff"],
+        "cx": ["#0f7cfb", "#f4f8ff"],
+        "ccx": ["#0f7cfb", "#f4f8ff"],
+        "mcx": ["#0f7cfb", "#f4f8ff"],
+        "cy": ["#0c63c9", "#f4f8ff"],
+        "cz": ["#009cb3", "#f4f8ff"],
+    },
+}
+
+
 def draw(
     gate: Callable[[Quant], None],
     num_qubits: int | list[int],
@@ -367,4 +404,4 @@ def draw(
         q = [p.alloc(num_qubits)]
     gate(*args, *q)
     p.execute()
-    return device.circuit.draw(**kwargs)
+    return device.circuit.draw(**kwargs, style=DRAW_STYLE)

{ket_lang-0.8.3.dev0 → ket_lang-0.8.4.dev0/src/ket_lang.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ket-lang
-Version: 0.8.3.dev0
+Version: 0.8.4.dev0
 Summary: Ket quantum programming language interpreter and library
 Home-page: https://quantumket.org
 Author: Evandro Chagas Ribeiro da Rosa