disable torch export test in docker (#3118)

* disable torch export test in docker * Run docker regression based on label --------- Co-authored-by: Matthias Reso <[email protected]>
pytorch · Apr 27, 2024 · 719526d · 719526d
1 parent 154ee49
commit 719526d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/.github/workflows/regression_tests_docker.yml b/.github/workflows/regression_tests_docker.yml
@@ -1,6 +1,9 @@
 name: Run Regression Tests on Docker
 
 on:
+  push:
+    tags:
+      - docker
   workflow_dispatch:
   # run every day at 5:15am
   schedule:

diff --git a/test/pytest/test_torch_export.py b/test/pytest/test_torch_export.py
@@ -49,6 +49,10 @@ def custom_working_directory(tmp_path):
     os.chdir(tmp_path)
 
 
+@pytest.mark.skipif(
+    os.environ.get("TS_RUN_IN_DOCKER", False),
+    reason="Test to be run outside docker",
+)
 @pytest.mark.skipif(PT_230_AVAILABLE == False, reason="torch version is < 2.3.0")
 def test_torch_export_aot_compile(custom_working_directory):
     # Get the path to the custom working directory
@@ -90,6 +94,10 @@ def test_torch_export_aot_compile(custom_working_directory):
     assert labels == EXPECTED_RESULTS
 
 
+@pytest.mark.skipif(
+    os.environ.get("TS_RUN_IN_DOCKER", False),
+    reason="Test to be run outside docker",
+)
 @pytest.mark.skipif(PT_230_AVAILABLE == False, reason="torch version is < 2.3.0")
 def test_torch_export_aot_compile_dynamic_batching(custom_working_directory):
     # Get the path to the custom working directory