ValueError: expected ndim=3, found ndim=2

Question

import os
import json
import tensorflow as tf
from sklearn.model_selection import train_test_split
import numpy as np
import matplotlib.pyplot as plt

data_path = &q