Update torch requirement from <2.1.0 to <2.2.0 in /tests

Updates the requirements on [torch](https://github.com/pytorch/pytorch) to permit the latest version.
- [Release notes](https://github.com/pytorch/pytorch/releases)
- [Changelog](https://github.com/pytorch/pytorch/blob/main/RELEASE.md)
- [Commits](pytorch/pytorch@ciflow/slow/0a9dbbbaadb86039ba17c1a11990f8a5e518ac06...v2.1.1)

---
updated-dependencies:
- dependency-name: torch
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <support@github.com>

tests/constraints.txt

@@ -22,4 +22,4 @@ pytest-html==3.2.0
 pytest-timeout==2.1.0
 jax<=0.4.14
 jaxlib<=0.4.14
-torch<2.1.0
+torch<2.2.0