From f6acf7075b3af97719edd3705f1268f0d357e0fa Mon Sep 17 00:00:00 2001
From: Alexander Hornburg <alexander.hornburg@amd.com>
Date: Wed, 23 Oct 2024 17:42:26 +0100
Subject: [PATCH] [Infra] support passing arguments to build_custom flow

---
 run-docker.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/run-docker.sh b/run-docker.sh
index b1fe44eb0c..1358337a37 100755
--- a/run-docker.sh
+++ b/run-docker.sh
@@ -142,7 +142,7 @@ elif [ "$1" = "build_custom" ]; then
   DOCKER_INTERACTIVE="-it"
   #FINN_HOST_BUILD_DIR=$BUILD_DATAFLOW_DIR/build
   gecho "Running build_custom: $BUILD_CUSTOM_DIR/$FLOW_NAME.py"
-  DOCKER_CMD="python -mpdb -cc -cq $FLOW_NAME.py"
+  DOCKER_CMD="python -mpdb -cc -cq $FLOW_NAME.py ${@:4}"
 elif [ -z "$1" ]; then
    gecho "Running container only"
    DOCKER_CMD="bash"