Audio Classification

rows = 3
cols = 3
n = rows * cols
fig, axes = plt.subplots(rows, cols, figsize=(10, 12))

for i, (audio, label) in enumerate(waveform_ds.take(n)):
r = i // cols
c = i % cols
ax = axes[r][c]
ax.plot(audio.numpy())
ax.set_yticks(np.arange(-1.2, 1.2, 0.2))
label = label.numpy().decode(‘utf-8’)
ax.set_title(label)

plt.show()

Gettin this error how can i solve
{{function_node _wrapped__IteratorGetNext_output_types_2_device/job:localhost/replica:0/task:0/device:CPU:0}} Can not squeeze dim[1], expected a dimension of 1, got 2
[[{{node Squeeze}}]] [Op:IteratorGetNext]

In order to expedite the trouble-shooting process, please provide a code snippet to reproduce the issue reported here. Thank you.