rust/lancedb
1
0
Fork 0
mirror of https://github.com/lancedb/lancedb.git synced 2026-01-15 08:12:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

make pandas an optional dependency in lancedb as well (#385)

Browse Source
This commit is contained in:
Chang She
2023-07-31 14:08:58 -04:00
committed by GitHub
parent cada35d5b7
commit c1f8feb6ed
10 changed files with 45 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
python/lancedb/common.py
View File

@@ -11,19 +11,18 @@
# See the License for the specific language governing permissions and
# limitations under the License.
from pathlib import Path
from typing import List, Union
from typing import Iterable, List, Union
import numpy as np
import pandas as pd
import pyarrow as pa
from .pydantic import LanceModel
from .util import safe_import_pandas
pd = safe_import_pandas()
DATA = Union[List[dict], dict, "pd.DataFrame", pa.Table, Iterable[pa.RecordBatch]]
VEC = Union[list, np.ndarray, pa.Array, pa.ChunkedArray]
URI = Union[str, Path]
# TODO support generator
DATA = Union[List[dict], List[LanceModel], dict, pd.DataFrame]
VECTOR_COLUMN_NAME = "vector"