diff --git a/agentic_security/probe_data/msj_data.py b/agentic_security/probe_data/msj_data.py index d54b078..0b75f61 100644 --- a/agentic_security/probe_data/msj_data.py +++ b/agentic_security/probe_data/msj_data.py @@ -1,6 +1,6 @@ from dataclasses import dataclass -from cache_to_disk import cache_to_disk +from cache_to_disk import cache_to_disk # noqa # TODO: refactor this class to use from .data diff --git a/tests/conftest.py b/tests/conftest.py index 927e0c8..19fe2dd 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -2,6 +2,7 @@ import os import pytest from cache_to_disk import delete_old_disk_caches + from agentic_security.logutils import logger