Context manager that times memory operations.
(self, node_id: str, operation_type: str, stage: str)
| 477 | |
| 478 | @contextmanager |
| 479 | def memory_timer(self, node_id: str, operation_type: str, stage: str): |
| 480 | """Context manager that times memory operations.""" |
| 481 | self.__init_timers__() |
| 482 | start_time = time.time() |
| 483 | try: |
| 484 | yield |
| 485 | finally: |
| 486 | end_time = time.time() |
| 487 | duration = (end_time - start_time) |
| 488 | self._timers[f"memory_{node_id}_{operation_type}_{stage}"] = duration |
| 489 | |
| 490 | def get_timer(self, timer_key: str) -> Optional[float]: |
| 491 | """Return the elapsed time recorded by the timer key.""" |
no test coverage detected