fix initializer

airbytehq · Nov 11, 2024 · 49a31e7 · 49a31e7
1 parent cdd696d
commit 49a31e7
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/unit_tests/destinations/vector_db_based/writer_test.py b/unit_tests/destinations/vector_db_based/writer_test.py
@@ -91,7 +91,13 @@ def test_write(omit_raw_text: bool):
     mock_indexer.post_sync.return_value = [post_sync_log_message]
 
     # Create the DestinationLangchain instance
-    writer = Writer(config_model, mock_indexer, mock_embedder, BATCH_SIZE, omit_raw_text)
+    writer = Writer(
+        config_model,
+        mock_indexer,
+        mock_embedder,
+        BATCH_SIZE,
+        omit_raw_text=omit_raw_text,
+    )
 
     output_messages = writer.write(configured_catalog, input_messages)
     output_message = next(output_messages)