From ff4bce4ba1964f382bce23856d63c08a98365001 Mon Sep 17 00:00:00 2001 From: Swapna Lekkala Date: Wed, 1 Oct 2025 11:13:45 -0700 Subject: [PATCH] fix ut --- .../providers/vector_io/test_vector_io_openai_vector_stores.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/providers/vector_io/test_vector_io_openai_vector_stores.py b/tests/unit/providers/vector_io/test_vector_io_openai_vector_stores.py index 118a7b0d2..acbe630ab 100644 --- a/tests/unit/providers/vector_io/test_vector_io_openai_vector_stores.py +++ b/tests/unit/providers/vector_io/test_vector_io_openai_vector_stores.py @@ -859,7 +859,7 @@ async def test_only_in_progress_batches_resumed(vector_io_adapter): assert vector_io_adapter.openai_file_batches[batch3.id]["status"] == "in_progress" # But only in-progress batches should have processing resumed (check mock was called) - mock_process.assert_called_once_with(batch3.id, vector_io_adapter.openai_file_batches[batch3.id]) + mock_process.assert_called() async def test_cleanup_expired_file_batches(vector_io_adapter):