22
33import pytest
44
5- from laygo . errors import ErrorHandler
6- from laygo . transformers . transformer import PipelineContext
7- from laygo . transformers . transformer import Transformer
5+ from laygo import ErrorHandler
6+ from laygo import PipelineContext
7+ from laygo import Transformer
88
99
1010class TestTransformerBasics :
@@ -216,7 +216,7 @@ def test_catch_with_error_isolation(self):
216216 errored_chunks = []
217217 transformer = Transformer .init (int , chunk_size = 1 ).catch (
218218 lambda t : t .map (lambda x : x / 0 ), # Division by zero
219- on_error = lambda chunk , error , context : errored_chunks .append (chunk ),
219+ on_error = lambda chunk , error , context : errored_chunks .append (chunk ), # type: ignore
220220 )
221221 result = list (transformer ([1 , 2 , 3 ]))
222222
@@ -244,7 +244,7 @@ def set_error_flag(_chunk, _error, context):
244244 Transformer .init (int , chunk_size = 1 )
245245 .catch (
246246 lambda t : t .map (lambda x : x / 0 ),
247- on_error = set_error_flag ,
247+ on_error = set_error_flag , # type: ignore
248248 )
249249 .short_circuit (lambda ctx : ctx .get ("error_occurred" , False ))
250250 )
@@ -266,7 +266,7 @@ def raise_on_error(ctx):
266266 Transformer .init (int , chunk_size = 1 )
267267 .catch (
268268 lambda t : t .map (lambda x : x / 0 ),
269- on_error = set_error_flag ,
269+ on_error = set_error_flag , # type: ignore
270270 )
271271 .short_circuit (raise_on_error )
272272 )
0 commit comments