From e4037f857870ef9d24370963eb5ac903587cda10 Mon Sep 17 00:00:00 2001 From: Eliott Rosenberg Date: Fri, 26 Jan 2024 09:41:33 -0800 Subject: [PATCH] types --- .../cirq/experiments/single_qubit_readout_calibration.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/cirq-core/cirq/experiments/single_qubit_readout_calibration.py b/cirq-core/cirq/experiments/single_qubit_readout_calibration.py index aebf9a0a834..34c18fcbdb6 100644 --- a/cirq-core/cirq/experiments/single_qubit_readout_calibration.py +++ b/cirq-core/cirq/experiments/single_qubit_readout_calibration.py @@ -81,7 +81,10 @@ def plot_heatmap( else: if ( - not len(axs) != 2 or type(axs[0]) != plt.Axes or type(axs[1]) != plt.Axes + not isinstance(axs, (tuple, list, np.ndarray)) + or len(axs) != 2 + or type(axs[0]) != plt.Axes + or type(axs[1]) != plt.Axes ): # pragma: no cover raise ValueError('axs should be a length-2 tuple of plt.Axes') # pragma: no cover for ax, title, data in zip(