From 9af342150693e6ae0854db27c32371456d76a535 Mon Sep 17 00:00:00 2001
From: nanocryk <6422796+nanocryk@users.noreply.github.com>
Date: Wed, 11 Dec 2024 16:03:56 +0100
Subject: [PATCH] update bench scripts

---
 tools/iforgor/tanssi.toml | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/tools/iforgor/tanssi.toml b/tools/iforgor/tanssi.toml
index 73d9216fa..7d0f14caa 100644
--- a/tools/iforgor/tanssi.toml
+++ b/tools/iforgor/tanssi.toml
@@ -98,6 +98,19 @@ name = "[Tanssi] Run dancebox benchmarks"
 only_in_dir = "**/tanssi"
 script = """
 export OUTPUT_PATH="tmp/dancebox_weights"
+export TEMPLATE_PATH="benchmarking/frame-weight-runtime-template.hbs"
+
+./tools/benchmarking.sh $1 "*"
+"""
+args = ["Which pallet to bench (* for all)"]
+
+[[entries]]
+name = "[Tanssi] Run flashbox benchmarks"
+only_in_dir = "**/tanssi"
+script = """
+export TEMPLATE_PATH=benchmarking/frame-weight-runtime-template.hbs
+export CHAIN=flashbox_dev
+export OUTPUT_PATH=tmp/flashbox_weights
 
 ./tools/benchmarking.sh $1 "*"
 """