Refactor deprecated torch.cuda.amp.autocast

[WenjieDu]

What does this PR do?

1. fixing FutureWarning: torch.cuda.amp.autocast(args...) is deprecated since torch 2.4 #520;

Before submitting

• Was this discussed/approved via a GitHub issue? Please add a link to it if that's the case.

[WenjieDu]

This PR is only for torch>=2.4, so hold it here for future merge.

[coveralls]

Pull Request Test Coverage Report for Build 13460517510

Warning: This coverage report may be inaccurate.

This pull request's base commit is no longer the HEAD commit of its target branch. This means it includes changes from outside the original pull request, including, potentially, unrelated coverage changes.

• For more information on this, see Tracking coverage changes with pull request builds.
• To avoid this issue with future PRs, see these Recommended CI Configurations.
• For a quick fix, rebase this PR at GitHub. Your next report should be accurate.

Details

• 4 of 6 (66.67%) changed or added relevant lines in 1 file are covered.
• 2 unchanged lines in 2 files lost coverage.
• Overall coverage increased (+0.01%) to 83.015%

---

Changes Missing Coverage	Covered Lines	Changed/Added Lines	%
pypots/nn/modules/reformer/local_attention.py	4	6	66.67%

Files with Coverage Reduction	New Missed Lines	%
pypots/forecasting/csdi/model.py	1	85.71%
pypots/imputation/csdi/model.py	1	85.29%

Totals
Change from base Build 13459953986:	0.01%
Covered Lines:	12336
Relevant Lines:	14860

---

💛 - Coveralls

[github-actions]

This pull request had no activity for 14 days. It will be closed in 1 week unless there is some new activity.

[github-actions]

This pull request had no activity for 14 days. It will be closed in 1 week unless there is some new activity.

[phro]

I am still getting the deprecation warning. I believe the version comparison is inverted. The condition should read torch.__version < "2.4", not >=.

[WenjieDu]

Right, we have fixed it in this commit ff94813. Thanks for reminding us, Jesse. We will release new version 0.12 with known bug fixes and new features later this week.