From a63d362af47be8650f3d6c3702165d630aca2e09 Mon Sep 17 00:00:00 2001
From: Madhur Tandon <madhur@outerbounds.co>
Date: Sun, 28 Apr 2024 18:48:52 +0530
Subject: [PATCH] fix script

---
 test/core/run_tests.py | 2 +-
 test_runner            | 5 ++++-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/test/core/run_tests.py b/test/core/run_tests.py
index 5f490a5fceb..a1b8cb27408 100644
--- a/test/core/run_tests.py
+++ b/test/core/run_tests.py
@@ -373,7 +373,7 @@ def run_test_cases(args):
     "--inherit-env", is_flag=True, default=False, help="Inherit env variables"
 )
 @click.option(
-    "--use-chaining-api",
+    "--use-chaining-api/--no-chaining-api",
     is_flag=True,
     default=False,
     help="Use low-level chaining API to generate commands.",
diff --git a/test_runner b/test_runner
index 14d4d2b3909..b712403f254 100755
--- a/test_runner
+++ b/test_runner
@@ -27,9 +27,12 @@ run_runtime_card_tests() {
   cd test/core && PYTHONPATH=`pwd`/../../ python3 run_tests.py --num-parallel 8 --contexts python3-all-local-cards-realtime --graphs $CARD_GRAPHS "$@" && cd ../../
 }
 
-if [[ "$1" == "chain" ]]; then
+if [[ "${1:-}" == "chain" ]]; then
   shift
   USE_CHAINING_API="--use-chaining-api"
+else
+  # Otherwise, set USE_CHAINING_API to "--no-chaining-api"
+  USE_CHAINING_API="--no-chaining-api"
 fi
 
 install_deps && install_extensions && run_tests "$USE_CHAINING_API" && run_runtime_card_tests "$USE_CHAINING_API"