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

fix import attr to use import attrs (#510)

Browse Source 
Thanks to #508, I used `attr` instead of the correct package `attrs`

s/attr/attrs
This commit is contained in:
Rob Meng
2023-09-23 00:30:56 -04:00
committed by GitHub
parent bc8670d7af
commit a695fb8030
3 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/lancedb/remote/__init__.py
View File

@@ -14,7 +14,7 @@
import abc
from typing import List, Optional
import attr
import attrs
import pyarrow as pa
from pydantic import BaseModel
@@ -44,7 +44,7 @@ class VectorQuery(BaseModel):
refine_factor: Optional[int] = None
@attr.define
@attrs.define
class VectorQueryResult:
# for now the response is directly seralized into a pandas dataframe
tbl: pa.Table

8
python/lancedb/remote/client.py
View File

@@ -16,7 +16,7 @@ import functools
from typing import Any, Callable, Dict, Optional, Union
import aiohttp
import attr
import attrs
import pyarrow as pa
from pydantic import BaseModel
@@ -43,14 +43,14 @@ async def _read_ipc(resp: aiohttp.ClientResponse) -> pa.Table:
return reader.read_all()
@attr.define(slots=False)
@attrs.define(slots=False)
class RestfulLanceDBClient:
db_name: str
region: str
api_key: Credential
host_override: Optional[str] = attr.field(default=None)
host_override: Optional[str] = attrs.field(default=None)
closed: bool = attr.field(default=False, init=False)
closed: bool = attrs.field(default=False, init=False)
@functools.cached_property
def session(self) -> aiohttp.ClientSession: