From ffd40768a7f4c711f8b311db327d954782f5d61d Mon Sep 17 00:00:00 2001
From: Jochen Hoenle <173445474+hoe-jo@users.noreply.github.com>
Date: Wed, 20 May 2026 15:56:05 +0200
Subject: [PATCH] [ai validator] expose batchsize to handle many requirements

---
 validation/ai_checker/ai_checker.bzl | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/validation/ai_checker/ai_checker.bzl b/validation/ai_checker/ai_checker.bzl
index e5be21f..d0816b7 100644
--- a/validation/ai_checker/ai_checker.bzl
+++ b/validation/ai_checker/ai_checker.bzl
@@ -85,8 +85,8 @@ def _run_ai_analysis(ctx, analysis_files, all_input_files, input_dirs, dep_dirs,
     if ctx.attr.model:
         args.add("--model", ctx.attr.model)
 
-    if ctx.attr._batch_size > 0:
-        args.add("--batch-size", str(ctx.attr._batch_size))
+    if ctx.attr.batch_size > 0:
+        args.add("--batch-size", str(ctx.attr.batch_size))
 
     # NOTE: --cache is intentionally NOT passed.  Bazel actions are
     # already cached by Bazel's action cache; an additional Python-level
@@ -182,7 +182,7 @@ _COMMON_AI_TEST_ATTRS = {
         doc = "Minimum average score required to pass the test (0-10).",
         default = "0.0",
     ),
-    "_batch_size": attr.int(
+    "batch_size": attr.int(
         doc = "Number of artefacts to process per batch (0 = all at once).",
         default = 0,
     ),