performance improvement

binref · Oct 23, 2024 · 54f1439 · 54f1439
1 parent b116729
commit 54f1439
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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: