def make_float_array(name: str, value: List[float]):
field = data_model_pb2.Field(name=name,
type=data_model_pb2.VECTOR,
value=np.array(value,
dtype=np.float32).tobytes())
return field
def make_binary_array(name: str, value: bytes):
field = data_model_pb2.Field(name=name,
type=data_model_pb2.VECTOR,
value=value)
return field
def make_float64_field(name: str, value: float):
field = data_model_pb2.Field(name=name,
type=data_model_pb2.DOUBLE,
value=np.float64(value).tobytes())
return field
def make_float32_field(name: str, value: float):
field = data_model_pb2.Field(name=name,
type=data_model_pb2.FLOAT,
value=np.float32(value).tobytes())
return field
def make_int64_field(name: str, value: int):
field = data_model_pb2.Field(name=name,
type=data_model_pb2.LONG,
value=np.int64(value).tobytes())
return field
def make_int32_field(name: str, value: int):
field = data_model_pb2.Field(name=name,
type=data_model_pb2.INT,
value=np.int32(value).tobytes())
return field
def make_string_field(name: str, value: str):
field = data_model_pb2.Field(name=name,
type=data_model_pb2.STRING,
value=bytes(value, encoding="utf8"))
return field