Skip to content

Commit 3e73bfe

Browse files
hoe-jocastler
authored andcommitted
[ai validator] expose batchsize to handle many requirements
1 parent 50be8b0 commit 3e73bfe

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

validation/ai_checker/ai_checker.bzl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -85,8 +85,8 @@ def _run_ai_analysis(ctx, analysis_files, all_input_files, input_dirs, dep_dirs,
8585
if ctx.attr.model:
8686
args.add("--model", ctx.attr.model)
8787

88-
if ctx.attr._batch_size > 0:
89-
args.add("--batch-size", str(ctx.attr._batch_size))
88+
if ctx.attr.batch_size > 0:
89+
args.add("--batch-size", str(ctx.attr.batch_size))
9090

9191
# NOTE: --cache is intentionally NOT passed. Bazel actions are
9292
# already cached by Bazel's action cache; an additional Python-level
@@ -182,7 +182,7 @@ _COMMON_AI_TEST_ATTRS = {
182182
doc = "Minimum average score required to pass the test (0-10).",
183183
default = "0.0",
184184
),
185-
"_batch_size": attr.int(
185+
"batch_size": attr.int(
186186
doc = "Number of artefacts to process per batch (0 = all at once).",
187187
default = 0,
188188
),

0 commit comments

Comments
 (0)