diff --git a/src/memray/_memray.pyx b/src/memray/_memray.pyx index f961cf699e..eb74474bf7 100644 --- a/src/memray/_memray.pyx +++ b/src/memray/_memray.pyx @@ -808,6 +808,7 @@ cdef class ProgressIndicator: def __exit__(self, type, value, traceback): if not self._report_progress: return + self._context_manager.update(self._task, completed=self._cumulative_num_processed) return self._context_manager.__exit__(type, value, traceback) cdef bool _time_for_refresh(self):