Fix issues in CI

[EnricoTrizio]

Description

Tests failing:

• General, PyTorch dismissed the _accumulate function we were using in data/datamodule.py

I added the function explicitly in the datamodule file

• On MacOS 14 because it doesn't support some device-related function anymore apparently

This closes #128

TODO

Check if tests succeed

Status

• Ready to go

[EnricoTrizio]

The problem should have been fixed in PyTorch from version 2.1 in pytorch/pytorch#111576
The test env installs PyTorch from conda-forge. Probably doing it from the PyTorch channel would give the latest pytorch where
the device issue has been solved

EDIT

I tried to use the latest version of PyTorch, as they apparently addressed the problem, but it didn't change much for us.
For now, I would test it on MacOS-13 and leave an open issue to remember to change it in the future when the 14 becomes stable again.

[codecov]

Codecov Report

Attention: Patch coverage is 80.00000% with 2 lines in your changes are missing coverage. Please review.

Project coverage is 91.97%. Comparing base (8c2ede9) to head (703f7fa).

Additional details and impacted files

[andrrizzi]

looks good!