Value Error occurs when I exercise your demo code

[htma]

Hi, my demo code as followed your example :
In [ ] : # -------------------------------------------------------------------------------
import numpy as np
import tensorflow as tf
from tensorflow.keras import Input
from tensorflow.keras.layers import Dense, LSTM
from tensorflow.keras.models import load_model, Model

from attention import AttentionNum_samples, time_steps, input_dim, output_dim = 100, 10, 1, 1
data_x = np.random.uniform(size=(num_samples, time_steps, input_dim))
data_y = np.random.uniform(size=(num_samples, output_dim))

In[ ]:

Define/compile the model.

model_input = Input(shape=(time_steps, input_dim))
model_input.shape

x = LSTM(32, return_sequences=True)(model_input)
x = Attention(units=32)(x)

------------------------------------------------------------------------------------------------------------------------

Here when I run above cells, an error always occurs:

[ValueError: Only input tensors may be passed as positional arguments. The following argument value should be passed as a keyword argument: None (of type <class 'NoneType'>)]

What can i Do it?
Thanks.