From dd022ad9bec631237b37cd7f1903aa6854f8f134 Mon Sep 17 00:00:00 2001
From: Kevin Lin <42618777+kevin314@users.noreply.github.com>
Date: Tue, 22 Jul 2025 23:00:03 -0500
Subject: [PATCH] [CI] Fix CI for pull request targets other than main (#632)

---
 .buildkite/pipeline.yml             | 6 ++++--
 fastvideo/v1/tests/modal/pr_test.py | 2 +-
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml
index 155d67704..863093c87 100644
--- a/.buildkite/pipeline.yml
+++ b/.buildkite/pipeline.yml
@@ -13,7 +13,7 @@ steps:
   - label: "Trigger Tests"
     plugins:
       - monorepo-diff#v1.4.0:
-          diff: "git diff --name-only $BUILDKITE_PULL_REQUEST_BASE_BRANCH...HEAD"
+          diff: 'git fetch origin "$BUILDKITE_PULL_REQUEST_BASE_BRANCH" && git diff --name-only origin/"$BUILDKITE_PULL_REQUEST_BASE_BRANCH"...HEAD'
           watch:
             - path: 
                 - "fastvideo/v1/models/encoders/**"
@@ -58,8 +58,10 @@ steps:
                   queue: "default"
             - path: 
                 - "fastvideo/v1/**/*.py"
+                - "pyproject.toml"
+                - "docker/Dockerfile.python3.12"
               config:
-                command: "timeout 30m .buildkite/scripts/pr_test.sh"
+                command: "timeout 45m .buildkite/scripts/pr_test.sh"
                 label: "SSIM Tests"
                 env:
                   - TEST_TYPE=ssim
diff --git a/fastvideo/v1/tests/modal/pr_test.py b/fastvideo/v1/tests/modal/pr_test.py
index 2bf879353..ed864bd05 100644
--- a/fastvideo/v1/tests/modal/pr_test.py
+++ b/fastvideo/v1/tests/modal/pr_test.py
@@ -74,7 +74,7 @@ def run_vae_tests():
 def run_transformer_tests():
     run_test("pytest ./fastvideo/v1/tests/transformers -vs")
 
-@app.function(gpu="L40S:2", image=image, timeout=1800)
+@app.function(gpu="L40S:2", image=image, timeout=2700)
 def run_ssim_tests():
     run_test("pytest ./fastvideo/v1/tests/ssim -vs")