diff --git a/tests/knowledge/custom_storage_knowledge_source_test.py b/tests/knowledge/custom_storage_knowledge_source_test.py index acd0c7cb8..6ed5b6280 100644 --- a/tests/knowledge/custom_storage_knowledge_source_test.py +++ b/tests/knowledge/custom_storage_knowledge_source_test.py @@ -4,7 +4,7 @@ import os import shutil import tempfile from pathlib import Path -from unittest.mock import patch, MagicMock +from unittest.mock import MagicMock, patch import pytest