Bug with an example code

[mmac3]

Hey, I've got an error using one of your examples and can't figure out the problem.

1) Confirm the below

• [✅] My Python version is 3.5 or higher
• [✅] I have searched through the issues Issues for a duplicate
• [✅] I've tested that my Keras model works as a stand-alone

2) Include the output of:

talos.__version__1.0 tensorflow.__version__2.4.0 keras.__version__2.4.3

3) Explain clearly what you expect to happen

I used exactly the same example from: https://autonomio.github.io/talos/#/Examples_Typical_Code

4) Explain what actually happened

I tried to run this code using a windows and linux environment. In both cases I got the same error (see below).

ValueError: Layer sequential_2 expects 1 input(s), but it received 2 input tensors. Inputs received: [<tf.Tensor 'IteratorGetNext:0' shape=(None, 4) dtype=float32>, <tf.Tensor 'IteratorGetNext:1' shape=(None, 3) dtype=float32>]

Does anyone know what the problem could be?

---