File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -269,7 +269,7 @@ async def test_write_documents_default_embedding(self) -> None:
269269 ):
270270 mock_client = AsyncMock ()
271271 mock_collection = AsyncMock ()
272- mock_batch = AsyncMock ()
272+ mock_batch = MagicMock ()
273273 mock_batch_context = AsyncMock ()
274274
275275 mock_client .collections .exists = AsyncMock (return_value = True )
@@ -318,7 +318,7 @@ async def test_write_documents_custom_embedding(self) -> None:
318318 ):
319319 mock_client = AsyncMock ()
320320 mock_collection = AsyncMock ()
321- mock_batch = AsyncMock ()
321+ mock_batch = MagicMock ()
322322 mock_batch_context = AsyncMock ()
323323
324324 mock_client .collections .exists = AsyncMock (return_value = False )
@@ -370,7 +370,7 @@ async def test_write_documents_ignores_per_write_embedding_with_warning(
370370 ):
371371 mock_client = AsyncMock ()
372372 mock_collection = AsyncMock ()
373- mock_batch = AsyncMock ()
373+ mock_batch = MagicMock ()
374374 mock_batch_context = AsyncMock ()
375375
376376 mock_client .collections .exists = AsyncMock (return_value = True )
You can’t perform that action at this time.
0 commit comments