Fix logging

[RolandBERTINJOHANNET]

batch functions were moved to utils but the logging code wasn't updated so no one can train a gw.

I changed the code to use the new locations. It runs fine locally but the pytest is unable to import shimmer.modules.utils I have no idea why

[bdvllrs]

This is only true if you use the last version of shimmer. But as it stands, the version in poetry.lock has not the batch_functions updated to the utils.

If you want to add this, we also need to update shimmer's dependency in poetry.

Most likely this only breaks your setup because you use shimmer inside this repo that you update independently from this one. Someone using poetry normally would not have the issue.

Also I'm closing because I was also working on it in the update_shimmer branch (which also contains the dependency updates).

[RolandBERTINJOHANNET]

ok