PTV/rainbowadn
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

remove generic from instrumentation

Browse Source
This commit is contained in:
AF 2023-01-13 15:33:49 +00:00
parent 49014b6a4d
commit d834c24e14
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
rainbowadn/instrument/instrumentation.py
View File

@ -1,13 +1,13 @@
import functools
from contextlib import ExitStack
from typing import Callable, Generic, TypeVar
from typing import Callable, TypeVar
__all__ = ('Instrumentation',)
IType = TypeVar('IType')
IType = TypeVar('IType', bound='Instrumentation')
class Instrumentation(Generic[IType]):
class Instrumentation:
deinstrumentation = {}
_method: Callable
_wrap: Callable
@ -24,8 +24,6 @@ class Instrumentation(Generic[IType]):
raise NotImplementedError
def __enter__(self: IType) -> IType:
if not isinstance(self, Instrumentation):
raise TypeError
if hasattr(self, '_method') or hasattr(self, '_wrap'):
raise RuntimeError
method = getattr(self.target, self.methodname)