From 54f143910f4348a6226ad3237d003ce4e31f090b Mon Sep 17 00:00:00 2001 From: jesko Date: Wed, 23 Oct 2024 20:14:39 +0200 Subject: [PATCH] performance improvement --- refinery/units/blockwise/__init__.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/refinery/units/blockwise/__init__.py b/refinery/units/blockwise/__init__.py index a302b2d0c0..fccdcfb6b0 100644 --- a/refinery/units/blockwise/__init__.py +++ b/refinery/units/blockwise/__init__.py @@ -187,10 +187,9 @@ def _mask(it): if not warnings: self.log_warn('additional warnings are suppressed') yield out - it = infinitize(it) if not masked: it = _mask(it) - return it + return infinitize(it) @abc.abstractmethod def operate(self, block, *args) -> int: