fixed linting

This commit is contained in:
Saifeddine ALOUI
2025-02-19 20:50:39 +01:00
parent 29cc12b56c
commit 45ee4dd08c
5 changed files with 5 additions and 1 deletions

View File

@@ -23,6 +23,7 @@ from ..base import BaseGraphStorage
if sys.platform.startswith("win"):
import asyncio.windows_events
asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
@@ -37,6 +38,7 @@ import psycopg
from psycopg.rows import namedtuple_row
from psycopg_pool import AsyncConnectionPool, PoolTimeout
class AGEQueryException(Exception):
"""Exception for the AGE queries."""

View File

@@ -28,6 +28,7 @@ from gremlin_python.driver import client, serializer
from gremlin_python.driver.aiohttp.transport import AiohttpTransport
from gremlin_python.driver.protocol import GremlinServerError
@final
@dataclass
class GremlinStorage(BaseGraphStorage):

View File

@@ -20,6 +20,7 @@ if not pm.is_installed("nano-vectordb"):
from nano_vectordb import NanoVectorDB
@final
@dataclass
class NanoVectorDBStorage(BaseVectorStorage):

View File

@@ -41,6 +41,7 @@ if not pm.is_installed("asyncpg"):
import asyncpg
from asyncpg import Pool
class PostgreSQLDB:
def __init__(self, config: dict[str, Any], **kwargs: Any):
self.host = config.get("host", "localhost")

View File

@@ -23,7 +23,6 @@ if not pm.is_installed("sqlalchemy"):
from sqlalchemy import create_engine, text
class TiDB:
def __init__(self, config, **kwargs):
self.host = config.get("host", None)