fix: isort linter issues!

dags/hivemind_etl_helpers/ingestion_pipeline.py

@@ -1,7 +1,7 @@
 import logging
 from datetime import datetime
-from dateutil.parser import parse
 
+from dateutil.parser import parse
 from hivemind_etl_helpers.src.utils.credentials import load_redis_credentials
 from hivemind_etl_helpers.src.utils.mongo import get_mongo_uri
 from hivemind_etl_helpers.src.utils.redis import RedisSingleton

dags/hivemind_etl_helpers/src/db/telegram/extract/__init__.py

@@ -1,2 +1,2 @@
-from .tc_chats import TelegramChats
 from .messages import ExtractMessages
+from .tc_chats import TelegramChats

dags/hivemind_etl_helpers/src/db/telegram/extract/messages.py

@@ -1,7 +1,7 @@
-from tc_neo4j_lib import Neo4jOps
 from datetime import datetime
 
 from hivemind_etl_helpers.src.db.telegram.schema import TelegramMessagesModel
+from tc_neo4j_lib import Neo4jOps
 
 
 class ExtractMessages:

dags/hivemind_etl_helpers/src/db/telegram/transform/messages.py

@@ -1,5 +1,5 @@
-from llama_index.core import Document
 from hivemind_etl_helpers.src.db.telegram.schema import TelegramMessagesModel
+from llama_index.core import Document
 
 
 class TransformMessages:

dags/hivemind_etl_helpers/src/db/telegram/utility.py

@@ -1,6 +1,4 @@
-from hivemind_etl_helpers.src.db.telegram.utils import (
-    TelegramPlatform,
-)
+from hivemind_etl_helpers.src.db.telegram.utils import TelegramPlatform
 
 
 class TelegramUtils(TelegramPlatform):

dags/hivemind_etl_helpers/src/db/telegram/utils/platform.py

@@ -1,6 +1,6 @@
-from bson import ObjectId
-
 from datetime import datetime, timezone
+
+from bson import ObjectId
 from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
 
 

dags/hivemind_etl_helpers/tests/integration/test_extract_messages.py

@@ -1,5 +1,5 @@
-from unittest import TestCase
 from datetime import datetime
+from unittest import TestCase
 
 from dotenv import load_dotenv
 from hivemind_etl_helpers.src.db.telegram.extract import ExtractMessages

dags/hivemind_etl_helpers/tests/integration/test_telegram_comminity.py

@@ -1,10 +1,9 @@
-from bson import ObjectId
-from unittest import TestCase
-
 from datetime import datetime
+from unittest import TestCase
 
-from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
+from bson import ObjectId
 from hivemind_etl_helpers.src.db.telegram.utils import TelegramPlatform
+from hivemind_etl_helpers.src.utils.mongo import MongoSingleton
 
 
 class TestTelegramPlatform(TestCase):

dags/hivemind_telegram_etl.py

@@ -5,7 +5,7 @@
 from airflow.decorators import task
 from dotenv import load_dotenv
 from hivemind_etl_helpers.ingestion_pipeline import CustomIngestionPipeline
-from hivemind_etl_helpers.src.db.telegram.extract import TelegramChats, ExtractMessages
+from hivemind_etl_helpers.src.db.telegram.extract import ExtractMessages, TelegramChats
 from hivemind_etl_helpers.src.db.telegram.transform import TransformMessages
 from hivemind_etl_helpers.src.db.telegram.utility import TelegramUtils