Improve pydantic integration (#384)

This commit is contained in:
Chang She
2023-07-31 12:16:44 -04:00
committed by GitHub
parent 2d25c263e9
commit cada35d5b7
9 changed files with 108 additions and 7 deletions

View File

@@ -17,11 +17,13 @@ import numpy as np
import pandas as pd
import pyarrow as pa
from .pydantic import LanceModel
VEC = Union[list, np.ndarray, pa.Array, pa.ChunkedArray]
URI = Union[str, Path]
# TODO support generator
DATA = Union[List[dict], dict, pd.DataFrame]
DATA = Union[List[dict], List[LanceModel], dict, pd.DataFrame]
VECTOR_COLUMN_NAME = "vector"