Fix import for ds

ngokchaoho · Dec 16, 2023 · a267f30 · a267f30
1 parent 485a4f0
commit a267f30
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_protocol.py b/tests/test_protocol.py
@@ -1,6 +1,7 @@
 import pytest
 
 from pyredis.protocol import extract_frame_from_buffer, encode_message
+from pyredis.datastore import DataStore
 from pyredis.types import (
     Array,
     BulkString,