Skip to content

Commit b1b49e2

Browse files
committed
disable torch export test in docker
1 parent 9a32216 commit b1b49e2

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

test/pytest/test_torch_export.py

+8
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,10 @@ def custom_working_directory(tmp_path):
4949
os.chdir(tmp_path)
5050

5151

52+
@pytest.mark.skipif(
53+
os.environ.get("TS_RUN_IN_DOCKER", False),
54+
reason="Test to be run outside docker",
55+
)
5256
@pytest.mark.skipif(PT_230_AVAILABLE == False, reason="torch version is < 2.3.0")
5357
def test_torch_export_aot_compile(custom_working_directory):
5458
# Get the path to the custom working directory
@@ -90,6 +94,10 @@ def test_torch_export_aot_compile(custom_working_directory):
9094
assert labels == EXPECTED_RESULTS
9195

9296

97+
@pytest.mark.skipif(
98+
os.environ.get("TS_RUN_IN_DOCKER", False),
99+
reason="Test to be run outside docker",
100+
)
93101
@pytest.mark.skipif(PT_230_AVAILABLE == False, reason="torch version is < 2.3.0")
94102
def test_torch_export_aot_compile_dynamic_batching(custom_working_directory):
95103
# Get the path to the custom working directory

0 commit comments

Comments
 (0)