VGAE model Line 82 in main.py

[chaitanya2903]

As far as I've understood, only two types of models are implemented - GNAE and VGNAE, but in line 82 of main.py there's an if condition which checks if args.model is 'VGAE' which I think is a typo. It should be

if args.model == 'VGNAE':
    loss = loss + (1 / data.num_nodes) * model.kl_loss()