diff --git a/mex/common/sinks/registry.py b/mex/common/sinks/registry.py index cc1ef6a7..d61d35cd 100644 --- a/mex/common/sinks/registry.py +++ b/mex/common/sinks/registry.py @@ -41,7 +41,7 @@ def load( ) -> Generator[Identifier, None, None]: """Load models to multiple sinks simultaneously.""" for sink, model_gen in zip( - self._sinks, tee(models, len(self._sinks)), strict=False + self._sinks, tee(models, len(self._sinks)), strict=True ): yield from sink.load(model_gen)