From b1c699aab54e4b7431a218c7c0fdaed98144194d Mon Sep 17 00:00:00 2001 From: Anastasia Kuporosova Date: Thu, 15 Feb 2024 14:13:28 +0100 Subject: [PATCH] try to bump up torch upperbound in tests (#22757) ### Details: - recently pytorch2.2 was [released](https://github.com/pytorch/pytorch/releases) ### Tickets: - *ticket-id* --- tests/constraints.txt | 2 +- tests/layer_tests/pytorch_tests/pytorch_layer_test_class.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/constraints.txt b/tests/constraints.txt index 13a2143dcac670..ca408c7090b9a4 100644 --- a/tests/constraints.txt +++ b/tests/constraints.txt @@ -24,4 +24,4 @@ pytest-html==4.1.1 pytest-timeout==2.2.0 jax<=0.4.14 jaxlib<=0.4.14 -torch>=1.13,<2.2 +torch>=1.13,<2.3 diff --git a/tests/layer_tests/pytorch_tests/pytorch_layer_test_class.py b/tests/layer_tests/pytorch_tests/pytorch_layer_test_class.py index cbdf9fc42179f8..438eeb8a355ac7 100644 --- a/tests/layer_tests/pytorch_tests/pytorch_layer_test_class.py +++ b/tests/layer_tests/pytorch_tests/pytorch_layer_test_class.py @@ -93,7 +93,7 @@ def use_torch_export(): from torch.fx.experimental.proxy_tensor import make_fx em = export(model, tuple(torch_inputs)) - if version.parse(torch.__version__) >= version.parse("2.2"): + if version.parse(torch.__version__) >= version.parse("2.3"): em = em.run_decompositions() print(em.graph_module.code)