|
3 | 3 | import collections.abc as collections |
4 | 4 | import numbers |
5 | 5 | import warnings |
6 | | -import weakref |
7 | 6 | from abc import ABCMeta, abstractmethod |
8 | 7 | from collections import OrderedDict |
9 | 8 | from typing import Any, Callable, Dict, List, Optional, Sequence, Tuple, Union |
@@ -262,34 +261,17 @@ def attach( |
262 | 261 | :class:`~ignite.engine.events.RemovableEventHandle`, which can be used to remove the handler. |
263 | 262 | """ |
264 | 263 | if isinstance(event_name, EventsList): |
265 | | - # Use weak reference to break circular reference: engine -> _event_handlers -> BaseLogger |
266 | | - # weak_self = weakref.ref(self) |
267 | | - |
268 | | - # def weak_log_handler(engine, event_name): |
269 | | - # logger_obj = weak_self() |
270 | | - # if logger_obj is not None: |
271 | | - # log_handler(engine, logger_obj, event_name) |
272 | | - |
273 | 264 | for name in event_name: |
274 | 265 | if name not in State.event_to_attr: |
275 | 266 | raise RuntimeError(f"Unknown event name '{name}'") |
276 | | - engine.add_event_handler(name, weak_log_handler, name) |
| 267 | + engine.add_event_handler(name, log_handler, name) |
277 | 268 |
|
278 | 269 | return RemovableEventHandle(event_name, log_handler, engine) |
279 | 270 |
|
280 | 271 | else: |
281 | 272 | if event_name not in State.event_to_attr: |
282 | 273 | raise RuntimeError(f"Unknown event name '{event_name}'") |
283 | 274 |
|
284 | | - # Use weak reference to break circular reference: engine -> _event_handlers -> BaseLogger |
285 | | - # weak_self = weakref.ref(self) |
286 | | - |
287 | | - # def weak_log_handler(engine): |
288 | | - # logger_obj = weak_self() |
289 | | - # if logger_obj is not None: |
290 | | - # log_handler(engine, logger_obj, event_name, *args, **kwargs) |
291 | | - |
292 | | - # return engine.add_event_handler(event_name, weak_log_handler) |
293 | 275 | return engine.add_event_handler(event_name, log_handler, self, event_name, *args, **kwargs) |
294 | 276 |
|
295 | 277 | def attach_output_handler(self, engine: Engine, event_name: Any, *args: Any, **kwargs: Any) -> RemovableEventHandle: |
|
0 commit comments