diff --git a/rainbowadn/instrument/instrumentation.py b/rainbowadn/instrument/instrumentation.py index e5f9c27..18147c5 100644 --- a/rainbowadn/instrument/instrumentation.py +++ b/rainbowadn/instrument/instrumentation.py @@ -22,7 +22,7 @@ class Instrumentation(Generic[IType]): raise NotImplementedError def __enter__(self: IType) -> IType: - self: Instrumentation + assert isinstance(self, Instrumentation) assert not hasattr(self, '_method') assert not hasattr(self, '_wrap') method = getattr(self.target, self.methodname) @@ -56,4 +56,5 @@ class Instrumentation(Generic[IType]): self.deinstrument() def enter(self: IType, stack: ExitStack) -> IType: + assert isinstance(self, Instrumentation) return stack.enter_context(self) diff --git a/setup.py b/setup.py new file mode 100644 index 0000000..ec5ba61 --- /dev/null +++ b/setup.py @@ -0,0 +1,12 @@ +from setuptools import setup + +setup( + name='rainbowadn', + version='', + packages=['rainbowadn.instrumentation'], + url='', + license='', + author='PARRRATE TNV', + author_email='', + description='' +)