File tree 1 file changed +8
-0
lines changed
1 file changed +8
-0
lines changed Original file line number Diff line number Diff line change @@ -49,6 +49,10 @@ def custom_working_directory(tmp_path):
49
49
os .chdir (tmp_path )
50
50
51
51
52
+ @pytest .mark .skipif (
53
+ os .environ .get ("TS_RUN_IN_DOCKER" , False ),
54
+ reason = "Test to be run outside docker" ,
55
+ )
52
56
@pytest .mark .skipif (PT_230_AVAILABLE == False , reason = "torch version is < 2.3.0" )
53
57
def test_torch_export_aot_compile (custom_working_directory ):
54
58
# Get the path to the custom working directory
@@ -90,6 +94,10 @@ def test_torch_export_aot_compile(custom_working_directory):
90
94
assert labels == EXPECTED_RESULTS
91
95
92
96
97
+ @pytest .mark .skipif (
98
+ os .environ .get ("TS_RUN_IN_DOCKER" , False ),
99
+ reason = "Test to be run outside docker" ,
100
+ )
93
101
@pytest .mark .skipif (PT_230_AVAILABLE == False , reason = "torch version is < 2.3.0" )
94
102
def test_torch_export_aot_compile_dynamic_batching (custom_working_directory ):
95
103
# Get the path to the custom working directory
You can’t perform that action at this time.
0 commit comments