Currently, the number of outputs of the Keras model is checked:

It'd be nice if the number of inputs could be checked too (and their shape/dtype). This should be possible according to the docs, which says a Keras Model has an .inputs attribute: https://keras.io/api/models/model/