You should be able to import MolecularMetrics and use it in trainer.py via:
...
edges_logits, nodes_logits = self.G(z)
MolecularMetrics.reconstruction_score(edges_logits, nodes_logits)

PS: you can post code snippets by wrapping them into three backticks ```, which makes debugging easier.