ctr-ll4 / src /utils /common.py
sanjin7's picture
Upload src/ with huggingface_hub
cea4a4b
raw
history blame contribute delete
295 Bytes
from typing import Any
import numpy as np
def element_to_list(a: Any) -> list:
if type(a) is not list:
return [a]
return a
def vectorise_dict(dt: dict, as_list=False) -> tuple | list:
keys, values = zip(*sorted(dt.items()))
return list(values) if as_list else values