Fix pylint complaint, #2612

coleifer · Aug 26, 2022 · 304e209 · 304e209
1 parent b383c6f
commit 304e209
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/peewee.py b/peewee.py
@@ -1246,7 +1246,7 @@ def __getitem__(self, item):
                                  'end-point.')
             return self.between(item.start, item.stop)
         return self == item
-    def __iter__(self): raise NotImplementedError  # Prevent accidents.
+    __iter__ = None  # Prevent infinite loop.
 
     def distinct(self):
         return NodeList((SQL('DISTINCT'), self))

diff --git a/tests/regressions.py b/tests/regressions.py
@@ -1740,4 +1740,4 @@ def assertNames(q, expected):
 
 class TestFunctionInfiniteLoop(BaseTestCase):
     def test_function_infinite_loop(self):
-        self.assertRaises(NotImplementedError, lambda: list(fn.COUNT()))
+        self.assertRaises(TypeError, lambda: list(fn.COUNT()))