diff --git a/tests/e2e/test_streaming.py b/tests/e2e/test_streaming.py
index e0fec5876..a2b80f381 100644
--- a/tests/e2e/test_streaming.py
+++ b/tests/e2e/test_streaming.py
@@ -1,5 +1,7 @@
 """E2E tests for streaming dataset functionality"""
 
+# pylint: disable=duplicate-code
+
 import pytest
 
 from axolotl.common.datasets import load_datasets