fixed linting
This commit is contained in:
parent
29cc12b56c
commit
45ee4dd08c
5 changed files with 5 additions and 1 deletions
|
|
@ -23,6 +23,7 @@ from ..base import BaseGraphStorage
|
||||||
|
|
||||||
if sys.platform.startswith("win"):
|
if sys.platform.startswith("win"):
|
||||||
import asyncio.windows_events
|
import asyncio.windows_events
|
||||||
|
|
||||||
asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
|
asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -37,6 +38,7 @@ import psycopg
|
||||||
from psycopg.rows import namedtuple_row
|
from psycopg.rows import namedtuple_row
|
||||||
from psycopg_pool import AsyncConnectionPool, PoolTimeout
|
from psycopg_pool import AsyncConnectionPool, PoolTimeout
|
||||||
|
|
||||||
|
|
||||||
class AGEQueryException(Exception):
|
class AGEQueryException(Exception):
|
||||||
"""Exception for the AGE queries."""
|
"""Exception for the AGE queries."""
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -28,6 +28,7 @@ from gremlin_python.driver import client, serializer
|
||||||
from gremlin_python.driver.aiohttp.transport import AiohttpTransport
|
from gremlin_python.driver.aiohttp.transport import AiohttpTransport
|
||||||
from gremlin_python.driver.protocol import GremlinServerError
|
from gremlin_python.driver.protocol import GremlinServerError
|
||||||
|
|
||||||
|
|
||||||
@final
|
@final
|
||||||
@dataclass
|
@dataclass
|
||||||
class GremlinStorage(BaseGraphStorage):
|
class GremlinStorage(BaseGraphStorage):
|
||||||
|
|
|
||||||
|
|
@ -20,6 +20,7 @@ if not pm.is_installed("nano-vectordb"):
|
||||||
|
|
||||||
from nano_vectordb import NanoVectorDB
|
from nano_vectordb import NanoVectorDB
|
||||||
|
|
||||||
|
|
||||||
@final
|
@final
|
||||||
@dataclass
|
@dataclass
|
||||||
class NanoVectorDBStorage(BaseVectorStorage):
|
class NanoVectorDBStorage(BaseVectorStorage):
|
||||||
|
|
|
||||||
|
|
@ -41,6 +41,7 @@ if not pm.is_installed("asyncpg"):
|
||||||
import asyncpg
|
import asyncpg
|
||||||
from asyncpg import Pool
|
from asyncpg import Pool
|
||||||
|
|
||||||
|
|
||||||
class PostgreSQLDB:
|
class PostgreSQLDB:
|
||||||
def __init__(self, config: dict[str, Any], **kwargs: Any):
|
def __init__(self, config: dict[str, Any], **kwargs: Any):
|
||||||
self.host = config.get("host", "localhost")
|
self.host = config.get("host", "localhost")
|
||||||
|
|
|
||||||
|
|
@ -23,7 +23,6 @@ if not pm.is_installed("sqlalchemy"):
|
||||||
from sqlalchemy import create_engine, text
|
from sqlalchemy import create_engine, text
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class TiDB:
|
class TiDB:
|
||||||
def __init__(self, config, **kwargs):
|
def __init__(self, config, **kwargs):
|
||||||
self.host = config.get("host", None)
|
self.host = config.get("host", None)
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue