From 1feca1b8c4911db00b42e9a54a09caaff4e51f45 Mon Sep 17 00:00:00 2001 From: bonnal-enzo Date: Mon, 4 Dec 2023 03:48:50 +0100 Subject: [PATCH] lint --- kioss/__init__.py | 2 +- kioss/_pipe.py | 12 +++++++----- 2 files changed, 8 insertions(+), 6 deletions(-) diff --git a/kioss/__init__.py b/kioss/__init__.py index 4a8cc42..1cfe52d 100644 --- 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 index ffa5874..f6fd95f 100644 --- a/kioss/_pipe.py +++ b/kioss/_pipe.py @@ -21,7 +21,9 @@ 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 @@ -29,13 +31,13 @@ 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