fix script

Netflix · Apr 28, 2024 · 7c3e5e5 · 7c3e5e5
1 parent 6463e4e
commit 7c3e5e5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git 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
@@ -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"