From a47473f3f454b0d092b34ad205e4ee5c26f634cc Mon Sep 17 00:00:00 2001 From: kiersten-stokes Date: Thu, 27 Jun 2024 15:03:05 -0500 Subject: [PATCH] Unpin pytest and remove TODO Signed-off-by: kiersten-stokes --- setup.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 4eedceb7..331f8649 100644 --- a/setup.py +++ b/setup.py @@ -16,9 +16,8 @@ # ORT 1.16 is not compatible: https://github.com/Xilinx/Vitis-AI/issues/1343 INSTALL_REQUIRE = ["optimum", "transformers>=4.38", "onnx", "onnxruntime-extensions"] -# TODO: unpin pytest once https://github.com/huggingface/transformers/pull/29154 is merged & released TESTS_REQUIRE = [ - "pytest<=7.4.4", + "pytest", "parameterized", "evaluate", "timm",