Skip to content

Commit b8825c8

Browse files
jeremymanningclaude
andcommitted
Fix variant filtering in consolidation script
The variant filter was using `_variant={variant}_` which failed to match directory names like `author_variant=content_tokenizer=...` because there's no underscore between the variant name and the next parameter. Changed to `variant={variant}` to correctly match all variant models. This fixes the issue where only 1 model (371 rows) was being consolidated instead of all 80 models (360,640 rows) for content-only variant. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
1 parent 912dc5f commit b8825c8

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

code/consolidate_model_results.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def consolidate_model_results(models_dir='models', output_path=None, save_csv=Fa
4545

4646
# Filter by variant
4747
if variant:
48-
model_dirs = [d for d in all_model_dirs if f'_variant={variant}_' in d.name]
48+
model_dirs = [d for d in all_model_dirs if f'variant={variant}' in d.name]
4949
variant_label = f"{variant} variant"
5050
else:
5151
model_dirs = [d for d in all_model_dirs if '_variant=' not in d.name]

0 commit comments

Comments
 (0)