From 5617400bde3cda522cfc8c7c023391feb44ed548 Mon Sep 17 00:00:00 2001
From: Kai Liao <140431279+kliao-csa@users.noreply.github.com>
Date: Thu, 2 May 2024 10:37:03 -0700
Subject: [PATCH] Update poll_quick_ci.py

---
 scripts/helpers/poll_quick_ci.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/scripts/helpers/poll_quick_ci.py b/scripts/helpers/poll_quick_ci.py
index 2ccaa6584b4a37..8cbf8dc8ddca46 100644
--- a/scripts/helpers/poll_quick_ci.py
+++ b/scripts/helpers/poll_quick_ci.py
@@ -1,9 +1,13 @@
+import argparse
 import logging
 import subprocess
 
 def main():
+  parser = argparse.ArgumentParser()
+  parser.add_argument("pr", help="Pull request number")
+  args = parser.parse_args()
   logging.info("Gathering info on checks being run for the current pull request.")
-  subprocess.run("gh pr checks -R project-chip/connectedhomeip ${{ github.event.number }} -L 500", shell=True)
+  subprocess.run(f"gh pr checks -R project-chip/connectedhomeip {args.pr} -L 500", shell=True)
 
 if __name__ == "__main__":
   main()