From 920dc1da2ff02b55e27a56851b996f29730f217e Mon Sep 17 00:00:00 2001 From: Alexander Myasoedov Date: Tue, 10 Dec 2024 15:40:32 +0200 Subject: [PATCH] fix(imports): --- agentic_security/probe_actor/test_fuzzer.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/agentic_security/probe_actor/test_fuzzer.py b/agentic_security/probe_actor/test_fuzzer.py index b345047..12eb8ca 100644 --- a/agentic_security/probe_actor/test_fuzzer.py +++ b/agentic_security/probe_actor/test_fuzzer.py @@ -1,11 +1,13 @@ import asyncio +from unittest.mock import AsyncMock, MagicMock, patch + import pytest -from unittest.mock import AsyncMock, patch, MagicMock -from agentic_security.models.schemas import ScanResult, Scan + +from agentic_security.models.schemas import Scan from agentic_security.probe_actor.fuzzer import ( - prompt_iter, - perform_single_shot_scan, perform_many_shot_scan, + perform_single_shot_scan, + prompt_iter, scan_router, )