Update

[ghstack-poisoned]
pytorch · Sep 16, 2024 · d9fda4a · d9fda4a
2 parents 3a99b43 + b1ff110
commit d9fda4a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/.github/unittest/linux/scripts/environment.yml b/.github/unittest/linux/scripts/environment.yml
@@ -19,6 +19,3 @@ dependencies:
     - coverage
     - h5py
     - orjson
-#    - onnx
-#    - onnxscript
-#    - onnxruntime
diff --git a/test/test_compile.py b/test/test_compile.py
@@ -8,8 +8,6 @@
 from pathlib import Path
 from typing import Any
 
-import onnxruntime
-
 import pytest
 
 import torch
@@ -797,6 +795,8 @@ def test_onnx_export_module(self, tmpdir):
 
         path = Path(tmpdir) / "file.onnx"
         onnx_program.save(str(path))
+        import onnxruntime
+
         ort_session = onnxruntime.InferenceSession(
             path, providers=["CPUExecutionProvider"]
         )
@@ -832,6 +832,8 @@ def test_onnx_export_seq(self, tmpdir):
 
         path = Path(tmpdir) / "file.onnx"
         onnx_program.save(str(path))
+        import onnxruntime
+
         ort_session = onnxruntime.InferenceSession(
             path, providers=["CPUExecutionProvider"]
         )