kristoferssolo/grovers-visualizer
1
0
Fork 0
mirror of https://github.com/kristoferssolo/grovers-visualizer.git synced 2025-10-21 20:10:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix(circle): value sign

Browse Source
This commit is contained in:
Kristofers Solo 2025-04-24 15:13:15 +03:00
parent 50e843a884
commit 64650dd7b0
Signed by: kristoferssolo
GPG Key ID: 74FF8144483D82C8
2 changed files with 3 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/grovers_visualizer/plot.py
View File

@ -9,7 +9,7 @@ from matplotlib.patches import Circle
from qiskit.quantum_info import Statevector
from .state import QubitState
from .utils import get_bar_color, is_optimal_iteration, sign
from .utils import get_bar_color, is_optimal_iteration
def plot_amplitudes(
@ -87,7 +87,7 @@ def plot_circle(
ax.text(
x,
y,
f"{prob * sign(y):.2f}",
f"{prob:.2f}",
color=color,
fontsize=10,
ha="left" if x >= 0 else "right",

6
src/grovers_visualizer/utils.py
View File

@ -1,6 +1,6 @@
from collections.abc import Iterator
from itertools import product
from math import copysign, floor, pi, sqrt
from math import floor, pi, sqrt
from .state import QubitState
@ -27,7 +27,3 @@ def get_bar_color(state: str, target_state: QubitState | None, iteration: int, o
if optimal_iteration and is_optimal_iteration(iteration, optimal_iteration):
return "green"
return "orange"
def sign(x: float) -> int:
return int(copysign(1, x))