Github Workflow

.github/workflows/checkTrainingParams.yml

@@ -0,0 +1,99 @@
+name: Check Training Parameters
+
+on:
+  pull_request:
+    branches: [ main ]
+    paths:
+      - 'XPointMLTest.py'  # Only trigger when this specific file changes
+
+jobs:
+  check-training-params:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout PR code
+        uses: actions/checkout@v3
+        with:
+          ref: ${{ github.event.pull_request.head.sha }}
+          path: pr-code
+          fetch-depth: 1
+
+      - name: Checkout main branch
+        uses: actions/checkout@v3
+        with:
+          ref: main
+          path: main-code
+
+      - name: Set up Python
+        uses: actions/setup-python@v4
+        with:
+          python-version: '3.10'
+
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install pytest
+
+      - name: Run parameter check
+        run: |
+          python - <<EOF
+import re
+import sys
+
+# Training parameter patterns to check
+params_to_check = {
+    'epochs': r'--epochs\', type=int, default=(\d+),',
+    'trainFrameFirst': r'--trainFrameFirst\', type=int, default=(\d+),',
+    'trainFrameLast': r'--trainFrameLast\', type=int, default=(\d+),',
+    'validationFrameFirst': r'--validationFrameFirst\', type=int, default=(\d+),',
+    'validationFrameLast': r'--validationFrameLast\', type=int, default=(\d+),',
+}
+
+# Files to check
+files_to_check = ['main-code/XPointMLTest.py', 'pr-code/XPointMLTest.py']
+
+main_params = {}
+pr_params = {}
+
+# Extract parameters from main branch
+with open('main-code/XPointMLTest.py', 'r') as f:
+    content = f.read()
+    for param, pattern in params_to_check.items():
+        match = re.search(pattern, content)
+        if match:
+            main_params[param] = int(match.group(1))
+        else:
+            print(f"Warning: Could not find parameter '{param}' in main branch code")
+
+# Extract parameters from PR
+with open('pr-code/XPointMLTest.py', 'r') as f:
+    content = f.read()
+    for param, pattern in params_to_check.items():
+        match = re.search(pattern, content)
+        if match:
+            pr_params[param] = int(match.group(1))
+        else:
+            print(f"Warning: Could not find parameter '{param}' in PR code")
+
+# Compare parameters
+mismatch = False
+for param in params_to_check.keys():
+    if param in main_params and param in pr_params:
+        if main_params[param] != pr_params[param]:
+            print(f"❌ Parameter '{param}' has changed: {main_params[param]} -> {pr_params[param]}")
+            mismatch = True
+        else:
+            print(f"✅ Parameter '{param}' unchanged: {main_params[param]}")
+    else:
+        print(f"⚠️ Could not compare '{param}' - missing from one or both branches")
+
+# Summary
+print("\n=== Parameter Check Summary ===")
+if mismatch:
+    print("❌ Training parameters have been modified!")
+    print("Detected changes to training configuration parameters.")
+    print("Please verify these changes are intentional and approved.")
+    sys.exit(1)
+else:
+    print("✅ All training parameters match the main branch!")
+    sys.exit(0)
+EOF