SungBeom's picture
Upload folder using huggingface_hub
4a51346
raw
history blame
316 Bytes
from clickhouse_connect.driver.options import np
from clickhouse_connect.driver.types import ByteSource
def read_numpy_array(source: ByteSource, np_type: str, num_rows: int):
dtype = np.dtype(np_type)
buffer = source.read_bytes(dtype.itemsize * num_rows)
return np.frombuffer(buffer, dtype, num_rows)