From f5b9ab51919f77a821ead5009326d816bc941bcd Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Mon, 21 Oct 2024 15:48:37 +0100 Subject: [PATCH] [CI] Fix versioning of h2d tests ghstack-source-id: faa425ca71953c1627690e08cf691683f78694d3 Pull Request resolved: https://github.com/pytorch/tensordict/pull/1053 --- benchmarks/common/h2d_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmarks/common/h2d_test.py b/benchmarks/common/h2d_test.py index 0e20aae75..b08298dc1 100644 --- a/benchmarks/common/h2d_test.py +++ b/benchmarks/common/h2d_test.py @@ -54,7 +54,7 @@ def default_device(): @pytest.mark.parametrize("consolidated", [False, True]) @pytest.mark.skipif( - TORCH_VERSION < version.parse("2.5.0"), reason="requires torch>=2.5" + TORCH_VERSION < version.parse("2.5.1"), reason="requires torch>=2.5" ) class TestTo: def test_to(self, benchmark, consolidated, td, default_device):