diff --git a/src/grovers_visualizer/plot.py b/src/grovers_visualizer/plot.py index e5788a5..76bddd6 100644 --- a/src/grovers_visualizer/plot.py +++ b/src/grovers_visualizer/plot.py @@ -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", diff --git a/src/grovers_visualizer/utils.py b/src/grovers_visualizer/utils.py index 419c7c6..4b7e82f 100644 --- a/src/grovers_visualizer/utils.py +++ b/src/grovers_visualizer/utils.py @@ -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))