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

windows paths

Browse Source
This commit is contained in:
Chang She
2023-05-04 11:41:05 -07:00
parent 3a2df0ce45
commit b6739f3f66
3 changed files with 19 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
python/lancedb/db.py
View File

@@ -18,6 +18,7 @@ import pyarrow as pa
from .common import URI, DATA
from .table import LanceTable
from .util import get_uri_scheme
class LanceDBConnection:
@@ -26,10 +27,12 @@ class LanceDBConnection:
"""
def __init__(self, uri: URI):
if isinstance(uri, str):
uri = Path(uri)
uri = uri.expanduser().absolute()
Path(uri).mkdir(parents=True, exist_ok=True)
is_local = isinstance(uri, Path) or get_uri_scheme(uri) == "file"
if is_local:
if isinstance(uri, str):
uri = Path(uri)
uri = uri.expanduser().absolute()
Path(uri).mkdir(parents=True, exist_ok=True)
self._uri = str(uri)
@property
@@ -43,7 +46,11 @@ class LanceDBConnection:
-------
A list of table names.
"""
return [p.stem for p in Path(self.uri).glob("*.lance")]
if get_uri_scheme(self.uri) == "file":
return [p.stem for p in Path(self.uri).glob("*.lance")]
raise NotImplementedError(
"List table_names is only supported for local filesystem for now"
)
def __len__(self) -> int:
return len(self.table_names())