diff --git a/src/calng/base_correction.py b/src/calng/base_correction.py index 97ec168f03cddc21f72ff5aa55a0deca0b9efb30..1f0e26c625f44908510bfa6ca072cc0c6c7f305d 100644 --- a/src/calng/base_correction.py +++ b/src/calng/base_correction.py @@ -570,13 +570,13 @@ class BaseCorrection(PythonDevice): **self._gpu_runner_init_args, ) - with self._buffer_lock: - for ( - constant, - data, - ) in self.calcat_friend.cached_constants.items(): - self.log_status_info(f"Reload constant {constant}") - self._load_constant_to_gpu(constant, data) + # TODO: lock to avoid dict changing size underneath us + for ( + constant, + data, + ) in self.calcat_friend.cached_constants.items(): + self.log_status_info(f"Reload constant {constant}") + self._load_constant_to_gpu(constant, data) self._has_updated_shapes = True