this is the correct code
test_image_generator = test_data_gen.flow_from_directory(batch_size=batch_size, target_size=(IMG_HEIGHT, IMG_HEIGHT), directory=PATH, classes=['test'], shuffle=False)