fix: make A/B tests work in release branches

Signed-off-by: Pablo Barbáchano <pablob@amazon.com>
pb8o · Dec 19, 2023 · 25126f9 · 25126f9
1 parent ee73ec4
commit 25126f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/framework/ab_test.py b/tests/framework/ab_test.py
@@ -40,7 +40,7 @@
 
 # Locally, this will always compare against main, even if we try to merge into, say, a feature branch.
 # We might want to do a more sophisticated way to determine a "parent" branch here.
-DEFAULT_A_REVISION = os.environ.get("BUILDKITE_PULL_REQUEST_BASE_BRANCH", "main")
+DEFAULT_A_REVISION = "origin/" + os.environ.get("BUILDKITE_PULL_REQUEST_BASE_BRANCH", "main")
 
 
 T = TypeVar("T")