mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-08-02 08:44:44 +00:00
fix typo in comment
Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
This commit is contained in:
parent
e50a546bc0
commit
54d28d13e9
1 changed files with 2 additions and 1 deletions
|
@ -81,7 +81,7 @@ class TestVectorStore:
|
||||||
"window_len, overlap_len, expected_chunks",
|
"window_len, overlap_len, expected_chunks",
|
||||||
[
|
[
|
||||||
(5, 2, 4), # Create 4 chunks with window of 5 and overlap of 2
|
(5, 2, 4), # Create 4 chunks with window of 5 and overlap of 2
|
||||||
(4, 1, 4), # Create 4 chunks with window of 3 and overlap of 1
|
(4, 1, 4), # Create 4 chunks with window of 4 and overlap of 1
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
def test_make_overlapped_chunks(self, window_len, overlap_len, expected_chunks):
|
def test_make_overlapped_chunks(self, window_len, overlap_len, expected_chunks):
|
||||||
|
@ -93,6 +93,7 @@ class TestVectorStore:
|
||||||
chunks = make_overlapped_chunks(document_id, text, window_len, overlap_len, original_metadata)
|
chunks = make_overlapped_chunks(document_id, text, window_len, overlap_len, original_metadata)
|
||||||
|
|
||||||
assert len(chunks) == expected_chunks
|
assert len(chunks) == expected_chunks
|
||||||
|
print(len(chunks), expected_chunks)
|
||||||
|
|
||||||
# Check that each chunk has the right metadata
|
# Check that each chunk has the right metadata
|
||||||
for chunk in chunks:
|
for chunk in chunks:
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue