Change net_type in TimeGAN from GRU to LSTM

[vandanajoshi2017]

I have run the TimeGAN model by the given example I want to change the network type parameter .by default it is taken GRU but I want LSTM
in the following code

def make_net(model, n_layers, hidden_units, output_units, net_type='GRU'):
if net_type=='GRU':
for i in range(n_layers):
model.add(GRU(units=hidden_units,
return_sequences=True,
name=f'GRU_{i + 1}'))