allowing test runs

Author: cquil11

.editorconfig

@@ -1,13 +1,21 @@
 name: "Run Sweep"
+run-name: Run Sweep - ${{ github.event.pull_request.title || github.ref_name }}
 
 on:
     workflow_dispatch:
     push:
         branches:
             - main
-            - diff-only-runs
-        # paths:
-        #     - "perf-changelog.yaml"
+        paths:
+            - "perf-changelog.yaml"
+    pull_request:
+        branches:
+            - main
+        types:
+            - ready_for_review
+            - synchronize
+        paths:
+            - "perf-changelog.yaml"
 
 jobs:
     get-jobs:
@@ -22,11 +30,18 @@ jobs:
 
             - id: get-jobs
               run: |
-                  pip install pydantic
+                  if [ "${{ github.event_name }}" == "pull_request" ]; then
+                      BASE_REF="origin/${{ github.base_ref }}"
+                      HEAD_REF="${{ github.event.pull_request.head.sha }}"
+                  else
+                      BASE_REF="${{ github.event.before }}"
+                      HEAD_REF="${{ github.event.after }}"
+                  fi
+
                   CONFIG_JSON=$(python3 ${GITHUB_WORKSPACE}/utils/process_changelog.py \
                       --changelog-file ${GITHUB_WORKSPACE}/perf-changelog.yaml \
-                      --base-ref origin/main \
-                      --head-ref ${{ github.sha }})
+                      --base-ref "$BASE_REF" \
+                      --head-ref "$HEAD_REF")
 
                   echo "search-space-config=$CONFIG_JSON" >> $GITHUB_OUTPUT
 

.github/workflows/run-sweep.yml

@@ -1,17 +1,12 @@
 import argparse
 import json
-from pprint import pprint
 import re
 import subprocess
 from collections import defaultdict
 
 import yaml
 from constants import GENERATE_SWEEPS_PY_SCRIPT, MASTER_CONFIGS, RUNNER_CONFIG
-from matrix_logic.generate_sweep_configs import (
-    seq_len_itos,
-    seq_len_stoi,
-    seq_len_to_str,
-)
+from matrix_logic.generate_sweep_configs import seq_len_to_str
 from matrix_logic.validation import load_config_files
 from pydantic import BaseModel, ConfigDict, Field
 
@@ -132,7 +127,6 @@ def main():
         else:
             final_results["single_node"][seq_len_str].append(result)
 
-    # pprint(final_results)
     print(json.dumps(final_results))