Merge pull request #79 from BatsResearch/modularize_static_batching

utils separation
BatsResearch · Jul 8, 2024 · 8abca96 · 8abca96
2 parents 4b1db48 + c4974f6
commit 8abca96
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/alfred/fm/utils.py b/alfred/fm/utils.py
@@ -541,7 +541,7 @@ def static_batch(queries: Query, batch_size: int = 1024) -> List[List[Query]]:
     batches = []
     batch = []
     for query in queries:
-        if len(batch) == batch_sz:
+        if len(batch) == batch_size:
             batches.append(batch)
             batch = []
         if isinstance(query, CompletionQuery):