foo

Author: SonSang

.github/workflows/alarm.yml

@@ -20,6 +20,7 @@ jobs:
 
     steps:
       - name: Download artifacts from triggering run
+        id: dl
         uses: actions/download-artifact@v4
         with:
           # Production에서 업로드한 아티팩트 이름 (고정 이름이라면 그대로 사용)
@@ -31,11 +32,11 @@ jobs:
 
       - name: Show downloaded files
         run: |
-          echo "Downloaded into ./artifacts"
-          ls -la ./artifacts || true
+          echo "Downloaded into ${{ steps.dl.outputs.download-path }}"
+          ls -la ${{ steps.dl.outputs.download-path }} || true
           echo
           echo "Tree:"
-          (command -v tree >/dev/null && tree -a ./artifacts) || true
+          (command -v tree >/dev/null && tree -a ${{ steps.dl.outputs.download-path }}) || true
 
       - name: Checkout base repo (default branch)
         uses: actions/checkout@v4
@@ -69,5 +70,7 @@ jobs:
           NO_CHANGE_PATIENCE: "100"
           RUNTIME_REGRESSION_TOLERANCE_PCT: "10"
           COMPILE_REGRESSION_TOLERANCE_PCT: "10"
+
+          ARTIFACTS_DIR: ${{ steps.dl.outputs.download-path }}
         run: |
           python test_wandb.py

test_wandb.py

@@ -93,15 +93,19 @@ def parse_speed_txt_lines(lines):
 
 # ---------- Benchmark results of the current PR ----------
 # Find a file that starts with "speed_test"
-artifacts_path = "./artifacts"
-current_txt_path = None
+artifacts_path = os.path.abspath(os.environ.get("ARTIFACTS_DIR", "./artifacts"))
+print("Artifacts path: ", artifacts_path)
 if not os.path.exists(artifacts_path):
     print("Could not find artifacts directory; skip.")
     sys.exit(0)
 
-for file in os.listdir(artifacts_path):
-    if file.startswith("speed_test") and file.endswith(".txt"):
-        current_txt_path = file
+current_txt_path = None
+for root, _, files in os.walk(artifacts_path):
+    for fname in files:
+        if fname.startswith("speed_test") and fname.endswith(".txt"):
+            current_txt_path = os.path.join(root, fname)
+            break
+    if current_txt_path:
         break
 
 if current_txt_path is None: