set heatmap bounds

unitaryfund · Jan 16, 2025 · 5c95219 · 5c95219
1 parent 5e59346
commit 5c95219
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/docs/source/examples/pt_zne.md b/docs/source/examples/pt_zne.md
@@ -186,7 +186,7 @@ print(circuit)
 ```
 ```{code-cell} ipython3
 ptmcnot = ptm_matrix(circuit, 2)
-ax = sns.heatmap(ptmcnot.real, linewidth=0.5)
+ax = sns.heatmap(ptmcnot.real, linewidth=0.5, vmin=-1, vmax=1)
 print("Ideal CNOT PTM")
 plt.show()
 ```
@@ -196,7 +196,7 @@ noisy_circuit_incoherent = circuit.with_noise(depolarize(p=0.3))
 print(noisy_circuit_incoherent)
 
 ptmcnot = ptm_matrix(noisy_circuit_incoherent, 2)
-ax = sns.heatmap(ptmcnot.real, linewidth=0.5)
+ax = sns.heatmap(ptmcnot.real, linewidth=0.5, vmin=-1, vmax=1)
 print("\n Pauli Transfer Matrix of noisy CNOT (incoherent)")
 plt.show()
 ```
@@ -206,7 +206,7 @@ noisy_circuit_coherent = circuit.with_noise(Ry(rads=np.pi/12))
 print(noisy_circuit_coherent)
 
 ptmcnot = ptm_matrix(noisy_circuit_coherent, 2)
-ax = sns.heatmap(ptmcnot.real, linewidth=0.5)
+ax = sns.heatmap(ptmcnot.real, linewidth=0.5, vmin=-1, vmax=1)
 print("Pauli Transfer Matrix of noisy CNOT (coherent)")
 plt.show()
 ```