Skip to content

Commit 147286d

Browse files
Fix last buffer not flushed
1 parent 248dadb commit 147286d

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

modules/embeddings-vector-indexes/pages/embeddings/cloud-providers.adoc

+3
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,9 @@ def main():
7676
movies_batch = []
7777
batch_n += 1
7878
79+
# Flush last batch
80+
import_batch(driver, movies_with_embeddings, batch_n)
81+
7982
# Import complete, show counters
8083
records, _, _ = driver.execute_query('''
8184
MATCH (m:Movie WHERE m.embedding IS NOT NULL)

modules/embeddings-vector-indexes/pages/embeddings/sentence-transformers.adoc

+3
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,9 @@ def main():
6666
movies_with_embeddings = []
6767
batch_n += 1
6868
69+
# Flush last batch
70+
import_batch(driver, movies_with_embeddings, batch_n)
71+
6972
# Import complete, show counters
7073
records, _, _ = driver.execute_query('''
7174
MATCH (m:Movie WHERE m.embedding IS NOT NULL)

0 commit comments

Comments
 (0)