Skip to content

Commit

Permalink
fix: isort linter issue!
Browse files Browse the repository at this point in the history
  • Loading branch information
amindadgar committed Nov 21, 2024
1 parent 66d8157 commit a5b85e7
Show file tree
Hide file tree
Showing 12 changed files with 12 additions and 12 deletions.
2 changes: 1 addition & 1 deletion dags/analyzer_helper/telegram/extract_raw_members.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,8 @@
DateTimeFormatConverter,
)
from github.neo4j_storage.neo4j_connection import Neo4jConnection
from tc_hivemind_backend.db.mongo import MongoSingleton
from pymongo import DESCENDING
from tc_hivemind_backend.db.mongo import MongoSingleton


class ExtractRawMembers:
Expand Down
2 changes: 1 addition & 1 deletion dags/hivemind_etl_helpers/github_etl.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,6 @@
from datetime import datetime

from dotenv import load_dotenv
from tc_hivemind_backend.ingest_qdrant import CustomIngestionPipeline
from hivemind_etl_helpers.src.db.github.extract import (
GithubExtraction,
fetch_issues,
Expand All @@ -13,6 +12,7 @@
)
from hivemind_etl_helpers.src.db.github.transform import GitHubTransformation
from llama_index.core import Document
from tc_hivemind_backend.ingest_qdrant import CustomIngestionPipeline


def process_github_vectorstore(
Expand Down
2 changes: 1 addition & 1 deletion dags/hivemind_etl_helpers/mediawiki_etl.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import logging

from tc_hivemind_backend.ingest_qdrant import CustomIngestionPipeline
from hivemind_etl_helpers.src.db.mediawiki.extractor import MediaWikiExtractor
from tc_hivemind_backend.ingest_qdrant import CustomIngestionPipeline


def process_mediawiki_etl(
Expand Down
2 changes: 1 addition & 1 deletion dags/hivemind_etl_helpers/notion_etl.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import logging

from tc_hivemind_backend.ingest_qdrant import CustomIngestionPipeline
from hivemind_etl_helpers.src.db.notion.extractor import NotionExtractor
from tc_hivemind_backend.ingest_qdrant import CustomIngestionPipeline


class NotionProcessor:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,9 +3,9 @@

from bson import ObjectId
from hivemind_etl_helpers.src.db.discord.discord_summary import DiscordSummary
from tc_hivemind_backend.db.mongo import MongoSingleton
from llama_index.core import Document, MockEmbedding, Settings
from llama_index.core.llms import MockLLM
from tc_hivemind_backend.db.mongo import MongoSingleton


class TestDiscordGroupedDataPreparation(TestCase):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -4,9 +4,9 @@
from hivemind_etl_helpers.src.db.discord.summary.prepare_summaries import (
PrepareSummaries,
)
from tc_hivemind_backend.db.mongo import MongoSingleton
from llama_index.core import MockEmbedding, Settings
from llama_index.core.llms import MockLLM
from tc_hivemind_backend.db.mongo import MongoSingleton


class TestPrepareSummaries(unittest.TestCase):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
import unittest
from unittest.mock import Mock

from tc_hivemind_backend.ingest_qdrant import CustomIngestionPipeline
from llama_index.core.ingestion import IngestionPipeline
from llama_index.core.schema import Document
from tc_hivemind_backend.ingest_qdrant import CustomIngestionPipeline


class TestIngestionPipeline(unittest.TestCase):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -6,9 +6,9 @@
from hivemind_etl_helpers.src.db.discord.discord_raw_message_to_document import (
discord_raw_to_documents,
)
from tc_hivemind_backend.db.mongo import MongoSingleton
from llama_index.core.indices.vector_store import VectorStoreIndex
from tc_hivemind_backend.db.credentials import load_postgres_credentials
from tc_hivemind_backend.db.mongo import MongoSingleton
from tc_hivemind_backend.db.pg_db_utils import setup_db
from tc_hivemind_backend.pg_vector_access import PGVectorAccess

Expand Down
2 changes: 1 addition & 1 deletion dags/hivemind_google_drive_etl.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,9 +5,9 @@

from airflow import DAG
from airflow.decorators import task
from tc_hivemind_backend.ingest_qdrant import CustomIngestionPipeline
from hivemind_etl_helpers.src.db.gdrive.gdrive_loader import GoogleDriveLoader
from hivemind_etl_helpers.src.utils.modules import ModulesGDrive
from tc_hivemind_backend.ingest_qdrant import CustomIngestionPipeline

with DAG(
dag_id="gdrive_vector_store",
Expand Down
2 changes: 1 addition & 1 deletion dags/hivemind_telegram_etl.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,6 @@
from airflow import DAG
from airflow.decorators import task
from dotenv import load_dotenv
from tc_hivemind_backend.ingest_qdrant import CustomIngestionPipeline
from hivemind_etl_helpers.src.db.telegram.extract import (
ExtractMessages,
ExtractMessagesDaily,
Expand All @@ -18,6 +17,7 @@
)
from hivemind_etl_helpers.src.db.telegram.utils import TelegramModules, TelegramPlatform
from qdrant_client.http import models
from tc_hivemind_backend.ingest_qdrant import CustomIngestionPipeline

# Common DAG configuration
default_args = {
Expand Down
2 changes: 1 addition & 1 deletion dags/violation_detection_helpers/extract.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
import logging
from datetime import datetime

from tc_hivemind_backend.db.mongo import MongoSingleton
from pymongo.cursor import Cursor
from tc_hivemind_backend.db.mongo import MongoSingleton


class ExtractPlatformRawData:
Expand Down
2 changes: 1 addition & 1 deletion dags/violation_detection_helpers/load.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
from typing import Any

from tc_hivemind_backend.db.mongo import MongoSingleton
from pymongo import UpdateOne
from tc_hivemind_backend.db.mongo import MongoSingleton


class LoadPlatformLabeledData:
Expand Down

0 comments on commit a5b85e7

Please sign in to comment.