get main_vX.Y truth data if running on main_vX.Y branch

dtcenter · Feb 22, 2023 · 63405c6 · 63405c6
1 parent 820c34c
commit 63405c6
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/.github/jobs/setup_and_run_diff.py b/.github/jobs/setup_and_run_diff.py
@@ -5,7 +5,7 @@
 import subprocess
 import shlex
 
-from docker_utils import VERSION_EXT
+from docker_utils import VERSION_EXT, get_branch_name
 
 ci_dir = os.path.join(os.environ.get('GITHUB_WORKSPACE'), '.github')
 sys.path.insert(0, ci_dir)
@@ -32,7 +32,11 @@
 if os.environ.get('GITHUB_EVENT_NAME') == "pull_request":
     output_data_branch = os.environ.get('GITHUB_BASE_REF')
 else:
-    output_data_branch = 'develop'
+    branch_name = get_branch_name()
+    if branch_name.startswith('main_v'):
+        output_data_branch = branch_name
+    else:
+        output_data_branch = 'develop'
 
 output_category = f"output-{output_data_branch}-{artifact_name}"