fix: missing import

quintoandar · Oct 14, 2024 · ff85d8b · ff85d8b
1 parent 9da3e48
commit ff85d8b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/unit/butterfree/clients/test_cassandra_client.py b/tests/unit/butterfree/clients/test_cassandra_client.py
@@ -1,6 +1,8 @@
 from typing import Any, Dict, List
 from unittest.mock import MagicMock
 
+import pytest
+
 from butterfree.clients import CassandraClient
 from butterfree.clients.cassandra_client import (
     EMPTY_STRING_HOST_ERROR,