File size: 1,168 Bytes
9c6594c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
# 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.
from typing import Union
def build_intlist(data: list[int]) -> list[int]:
return data
def build_tensorlist(data: list[object]) -> list[object]:
return data
def build_doublelist(data: list[float]) -> list[float]:
return data
def build_boollist(data: list[bool]) -> list[bool]:
return data
def build_tensor_from_id(data: Union[int, object]) -> Union[int, None]:
if isinstance(data, int):
# just the id, can't really do anything
return data
return None
def restore_type_tag(value: object, type_str: str) -> object:
# 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
|