types

eliottrosenberg · Jan 26, 2024 · e4037f8 · e4037f8
1 parent d6cc9cb
commit e4037f8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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(