| # 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 | |