Traktor/myenv/Lib/site-packages/torch/package/_mock.py
2024-05-26 05:12:46 +02:00

123 lines
2.8 KiB
Python

_magic_methods = [
"__subclasscheck__",
"__hex__",
"__rmul__",
"__float__",
"__idiv__",
"__setattr__",
"__div__",
"__invert__",
"__nonzero__",
"__rshift__",
"__eq__",
"__pos__",
"__round__",
"__rand__",
"__or__",
"__complex__",
"__divmod__",
"__len__",
"__reversed__",
"__copy__",
"__reduce__",
"__deepcopy__",
"__rdivmod__",
"__rrshift__",
"__ifloordiv__",
"__hash__",
"__iand__",
"__xor__",
"__isub__",
"__oct__",
"__ceil__",
"__imod__",
"__add__",
"__truediv__",
"__unicode__",
"__le__",
"__delitem__",
"__sizeof__",
"__sub__",
"__ne__",
"__pow__",
"__bytes__",
"__mul__",
"__itruediv__",
"__bool__",
"__iter__",
"__abs__",
"__gt__",
"__iadd__",
"__enter__",
"__floordiv__",
"__call__",
"__neg__",
"__and__",
"__ixor__",
"__getitem__",
"__exit__",
"__cmp__",
"__getstate__",
"__index__",
"__contains__",
"__floor__",
"__lt__",
"__getattr__",
"__mod__",
"__trunc__",
"__delattr__",
"__instancecheck__",
"__setitem__",
"__ipow__",
"__ilshift__",
"__long__",
"__irshift__",
"__imul__",
"__lshift__",
"__dir__",
"__ge__",
"__int__",
"__ior__",
]
class MockedObject:
_name: str
def __new__(cls, *args, **kwargs):
# _suppress_err is set by us in the mocked module impl, so that we can
# construct instances of MockedObject to hand out to people looking up
# module attributes.
# Any other attempt to construct a MockedObject instance (say, in the
# unpickling process) should give an error.
if not kwargs.get("_suppress_err"):
raise NotImplementedError(
f"Object '{cls._name}' was mocked out during packaging "
f"but it is being used in '__new__'. If this error is "
"happening during 'load_pickle', please ensure that your "
"pickled object doesn't contain any mocked objects."
)
# Otherwise, this is just a regular object creation
# (e.g. `x = MockedObject("foo")`), so pass it through normally.
return super().__new__(cls)
def __init__(self, name: str, _suppress_err: bool):
self.__dict__["_name"] = name
def __repr__(self):
return f"MockedObject({self._name})"
def install_method(method_name):
def _not_implemented(self, *args, **kwargs):
raise NotImplementedError(
f"Object '{self._name}' was mocked out during packaging but it is being used in {method_name}"
)
setattr(MockedObject, method_name, _not_implemented)
for method_name in _magic_methods:
install_method(method_name)