| # mypy: allow-untyped-defs |
| # These functions are referenced from the pickle archives produced by |
| # ScriptModule.save() |
| |
| |
| # These (`build_*`) functions used to be used by `pickler.cpp` to specify |
| # the type of the list for certain special types, but now all lists get |
| # a type attached and restored via `restore_type_tag` below. The legacy |
| # functions should stick around for backwards-compatibility. |
| |
| |
| def build_intlist(data): |
| return data |
| |
| |
| def build_tensorlist(data): |
| return data |
| |
| |
| def build_doublelist(data): |
| return data |
| |
| |
| def build_boollist(data): |
| return data |
| |
| |
| def build_tensor_from_id(data): |
| if isinstance(data, int): |
| # just the id, can't really do anything |
| return data |
| |
| |
| def restore_type_tag(value, type_str): |
| # The type_ptr is used by the jit unpickler to restore the full static type |
| # to container types like list when they are re-loaded, but this doesn't |
| # matter for Python, so just return the plain value |
| return value |
