We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d4168b2 commit e3f1b8cCopy full SHA for e3f1b8c
src/pytorch_lightning/profilers/advanced.py
@@ -17,7 +17,7 @@
17
import logging
18
import pstats
19
from pathlib import Path
20
-from typing import Dict, Optional, Union
+from typing import Dict, Optional, Tuple, Union
21
22
from pytorch_lightning.profilers.profiler import Profiler
23
@@ -82,7 +82,7 @@ def teardown(self, stage: Optional[str] = None) -> None:
82
super().teardown(stage=stage)
83
self.profiled_actions = {}
84
85
- def __reduce__(self):
+ def __reduce__(self) -> Tuple:
86
# avoids `TypeError: cannot pickle 'cProfile.Profile' object`
87
return (
88
self.__class__,
0 commit comments