lint

ebonnal · Dec 4, 2023 · 1feca1b · 1feca1b
1 parent d76c790
commit 1feca1b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/kioss/__init__.py b/kioss/__init__.py
@@ -3,5 +3,5 @@
 from kioss import _pipe
 from kioss._visit import _explanation, _iter_production
 
-_pipe.ITERATOR_PRODUCING_VISITOR = _iter_production.IteratorProducingVisitor()
+_pipe.ITERATOR_PRODUCING_VISITOR_CLASS = _iter_production.IteratorProducingVisitor
 _pipe.EXPLAINING_VISITOR_CLASS = _explanation.ExplainingVisitor
diff --git a/kioss/_pipe.py b/kioss/_pipe.py
@@ -21,21 +21,23 @@
 U = TypeVar("U")
 R = TypeVar("R")
 
-ITERATOR_PRODUCING_VISITOR: "Optional[_iter_production.IteratorProducingVisitor]" = None
+ITERATOR_PRODUCING_VISITOR_CLASS: "Optional[Type[_iter_production.IteratorProducingVisitor]]" = (
+    None
+)
 EXPLAINING_VISITOR_CLASS: "Optional[Type[_explanation.ExplainingVisitor]]" = None
 
 
 class APipe(Iterable[T], ABC):
     upstream: "Optional[APipe]"
 
     def __iter__(self) -> Iterator[T]:
-        if ITERATOR_PRODUCING_VISITOR is None:
-            raise ValueError("_plan.ITERATOR_PRODUCING_VISITOR is None")
-        return self._accept(ITERATOR_PRODUCING_VISITOR)
+        if ITERATOR_PRODUCING_VISITOR_CLASS is None:
+            raise ValueError("_pipe.ITERATOR_PRODUCING_VISITOR_CLASS is None")
+        return self._accept(ITERATOR_PRODUCING_VISITOR_CLASS())
 
     def __repr__(self) -> str:
         if EXPLAINING_VISITOR_CLASS is None:
-            raise ValueError("_plan.EXPLAINER_VISITOR is None")
+            raise ValueError("_pipe.EXPLAINING_VISITOR_CLASS is None")
         return self._accept(EXPLAINING_VISITOR_CLASS())
 
     @abstractmethod