From 0ec3a8f5e460d2c724d975a16451baf04e4440af Mon Sep 17 00:00:00 2001 From: henri123lemoine Date: Tue, 27 Jun 2023 07:50:16 -0400 Subject: [PATCH] Reorder imports --- src/dataset/pinecone_db_handler.py | 2 +- src/dataset/sql_db_handler.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/dataset/pinecone_db_handler.py b/src/dataset/pinecone_db_handler.py index f5d7187..07c3286 100644 --- a/src/dataset/pinecone_db_handler.py +++ b/src/dataset/pinecone_db_handler.py @@ -1,8 +1,8 @@ # dataset/pinecone_db_handler.py +import os import json import pinecone -import os from .settings import PINECONE_INDEX_NAME, PINECONE_VALUES_DIMS, PINECONE_METRIC, PINECONE_METADATA_ENTRIES diff --git a/src/dataset/sql_db_handler.py b/src/dataset/sql_db_handler.py index 008a252..d48d46d 100644 --- a/src/dataset/sql_db_handler.py +++ b/src/dataset/sql_db_handler.py @@ -1,7 +1,7 @@ # dataset/sql_db_handler.py +from typing import List, Dict, Union import sqlite3 -from typing import List, Dict, Any from .settings import SQL_DB_PATH @@ -52,7 +52,7 @@ class SQLDB: except sqlite3.Error as e: logger.error(f"The error '{e}' occurred.") - def upsert_entry(self, entry: Dict[str, Any]) -> bool: + def upsert_entry(self, entry: Dict[str, Union[str, list]]) -> bool: with sqlite3.connect(self.db_name) as conn: cursor = conn.cursor() try: