ValueError in TxGNN.py

[tgm-mjabri]

Why in line 679 we're unpacking 3 values:
_, scores, _ = self.retrieve_gates_scores_penalties()
while the retrieve_gates_scores_penalties function only return 2 items in line 676
return whole_graph, test_graph

P.S: shouldn't the retrieve_save_gates take relation as an argument as well ?