From bc4694802cf0dfcfa7a8498c80a7487837eddeb6 Mon Sep 17 00:00:00 2001 From: whiterabbit1983 Date: Tue, 15 Oct 2024 09:53:46 +0000 Subject: [PATCH] refactor: Lint agents-api (CI) --- agents-api/tests/test_activities.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/agents-api/tests/test_activities.py b/agents-api/tests/test_activities.py index 6168c611c..b8fec644c 100644 --- a/agents-api/tests/test_activities.py +++ b/agents-api/tests/test_activities.py @@ -1,5 +1,5 @@ -from uuid import uuid4 from unittest.mock import patch +from uuid import uuid4 from ward import test @@ -66,9 +66,10 @@ async def _( cozo_client, max_batch_size=2, ) - + assert embed_query.call_count == 3 + @test("activity: call demo workflow via temporal client") async def _(): async with patch_testing_temporal() as (_, mock_get_client):