Fix CI tests

noxfile.py

@@ -30,8 +30,8 @@
     "typing_extensions",
     "ml-dtypes",
 )
-ONNX = "onnx==1.16"
-ONNX_RUNTIME = "onnxruntime==1.17.1"
+ONNX = "onnx==1.17"
+ONNX_RUNTIME = "onnxruntime==1.20.1"
 PYTORCH = "torch==2.3.1"
 TORCHVISON = "torchvision==0.18.1"
 TRANSFORMERS = "transformers==4.37.2"
@@ -104,6 +104,7 @@ def test_ort_nightly(session):
         PYTORCH,
         TORCHVISON,
         ONNX,
+        TRANSFORMERS,
         *ONNX_RUNTIME_NIGHTLY_DEPENDENCIES,
     )
     session.install("-r", "requirements/ci/requirements-ort-nightly.txt")
@@ -132,32 +133,3 @@ def test_experimental_torchlib_tracing(session):
         *session.posargs,
         env={"TORCHLIB_EXPERIMENTAL_PREFER_TRACING": "1"},
     )
-
-
-@nox.session(tags=["test-dort"])
-def test_dort(session):
-    """Test the conversion of a couple of models from transformers."""
-    session.install(
-        *COMMON_TEST_DEPENDENCIES,
-    )
-    torch_version, transformers_version = session.posargs
-
-    if torch_version == "nightly":
-        session.install(
-            "--pre",
-            "torch",
-            "torchvision",
-            "torchaudio",
-            "--index-url",
-            "https://download.pytorch.org/whl/nightly/cpu",
-        )
-    else:
-        session.install("torch", "torchvision", "torchaudio")
-
-    session.install("torch", "torchvision", "torchaudio")
-    session.install(f"transformers=={transformers_version}")
-    session.install("onnxruntime-training==1.17.1")
-
-    session.run("pip", "list")
-    session.run("pytest", "onnxscript")
-    session.run("pytest", "tests")

requirements-dev.txt

@@ -1,5 +1,5 @@
 setuptools>=61.0.0
-numpy<2.0
+numpy
 onnx-weekly>=1.17.0.dev20240325
 onnxruntime>=1.17.0
 typing_extensions
@@ -30,8 +30,8 @@ pytest-subtests
 pytest-xdist
 pytest!=7.1.0
 pyyaml
-torch>=2.1
-torchvision>=0.16.0
+torch>=2.3
+torchvision>=0.18.0
 transformers>=4.37.2
 
 # Lint