fix: Send batch info to graph_norm

Open Ramlaoui opened this issue 10 months ago • 0 comments

Similar to https://github.com/RolnickLab/ocp/pull/68. Unused imports were also removed.

Feb 12 '25 17:02 Ramlaoui