pytorch/test/package/package_c/__init__.py
Lillian Johnson e27f861db7 [torch.Package/TorchScript] TS into torch.Package test cases (#54894)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/54894

Test cases to test torch.Package's handling of TorchScript objects.

TODO: test mapping storages to different device

Test Plan: Imported from OSS

Reviewed By: suo

Differential Revision: D27832544

Pulled By: Lilyjjo

fbshipit-source-id: 6a67938a428b57520fead698da1412623ece9dbd
2021-05-14 08:21:44 -07:00

12 lines
174 B
Python

result = "package_c"
class PackageCObject:
__slots__ = ["obj"]
def __init__(self, obj):
self.obj = obj
def return_result(self):
return result