error in the tutorial of MPNN

[Jameel9]

Issue Type

Bug

Source

source

Keras Version

3.4.1

Custom Code

Yes

OS Platform and Distribution

No response

Python version

No response

GPU model and memory

No response

Current Behavior?

I just tried to run the tutorial on colab and I got the following error.

Standalone code to reproduce the issue or tutorial link

https://keras.io/examples/graph/mpnn-molecular-graphs/

Relevant log output

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-41-f21d54292481> in <cell line: 33>()
     31 
     32 
---> 33 mpnn = MPNNModel(
     34     atom_dim=x_train[0][0][0].shape[0],
     35     bond_dim=x_train[1][0][0].shape[0],

3 frames
/usr/local/lib/python3.10/dist-packages/keras/src/backend/common/variables.py in standardize_shape(shape)
    528             raise ValueError("Undefined shapes are not supported.")
    529         if not hasattr(shape, "__iter__"):
--> 530             raise ValueError(f"Cannot convert '{shape}' to a shape.")
    531         if config.backend() == "tensorflow":
    532             if isinstance(shape, tf.TensorShape):

ValueError: Cannot convert '29' to a shape.