revise GNNLux temporal layers

[CarloLucibello]

As done in #560 for the Flux counterpart

[juanmigutierrez]

I am following the resolve of this issue in the following commit #595. Open to recieve any comments and steps of what is expected specifically !