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

add LanceModel to docs (#386)

Browse Source 
Co-authored-by: Chang She <chang@lancedb.com>
This commit is contained in:
Chang She
2023-07-31 15:12:02 -04:00
committed by GitHub
parent b06e214d29
commit 4b45128bd6
3 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
python/lancedb/pydantic.py
View File

@@ -275,10 +275,16 @@ class LanceModel(pydantic.BaseModel):
@classmethod
def to_arrow_schema(cls):
"""
Get the Arrow Schema for this model.
"""
return pydantic_to_schema(cls)
@classmethod
def field_names(cls) -> List[str]:
"""
Get the field names of this model.
"""
if PYDANTIC_VERSION.major < 2:
return list(cls.__fields__.keys())
return list(cls.model_fields.keys())