format: sort imports

dworthen · dworthen · commit ff97a14ef1de · 2025-10-22T11:37:53.000-07:00
diff --git a/docs/examples_notebooks/api_overview.ipynb b/docs/examples_notebooks/api_overview.ipynb
@@ -28,9 +28,8 @@
     "from pathlib import Path\n",
     "from pprint import pprint\n",
     "\n",
-    "import pandas as pd\n",
-    "\n",
     "import graphrag.api as api\n",
+    "import pandas as pd\n",
     "from graphrag.config.load_config import load_config\n",
     "from graphrag.index.typing.pipeline_run_result import PipelineRunResult"
    ]
diff --git a/docs/examples_notebooks/custom_vector_store.ipynb b/docs/examples_notebooks/custom_vector_store.ipynb
@@ -60,7 +60,6 @@
     "\n",
     "import numpy as np\n",
     "import yaml\n",
-    "\n",
     "from graphrag.config.models.vector_store_schema_config import VectorStoreSchemaConfig\n",
     "from graphrag.data_model.types import TextEmbedder\n",
     "\n",
diff --git a/docs/examples_notebooks/drift_search.ipynb b/docs/examples_notebooks/drift_search.ipynb
@@ -20,7 +20,6 @@
     "from pathlib import Path\n",
     "\n",
     "import pandas as pd\n",
-    "\n",
     "from graphrag.config.enums import ModelType\n",
     "from graphrag.config.models.drift_search_config import DRIFTSearchConfig\n",
     "from graphrag.config.models.language_model_config import LanguageModelConfig\n",
diff --git a/docs/examples_notebooks/global_search.ipynb b/docs/examples_notebooks/global_search.ipynb
@@ -19,7 +19,6 @@
     "import os\n",
     "\n",
     "import pandas as pd\n",
-    "\n",
     "from graphrag.config.enums import ModelType\n",
     "from graphrag.config.models.language_model_config import LanguageModelConfig\n",
     "from graphrag.language_model.manager import ModelManager\n",
diff --git a/docs/examples_notebooks/global_search_with_dynamic_community_selection.ipynb b/docs/examples_notebooks/global_search_with_dynamic_community_selection.ipynb
@@ -19,7 +19,6 @@
     "import os\n",
     "\n",
     "import pandas as pd\n",
-    "\n",
     "from graphrag.config.enums import ModelType\n",
     "from graphrag.config.models.language_model_config import LanguageModelConfig\n",
     "from graphrag.language_model.manager import ModelManager\n",
diff --git a/docs/examples_notebooks/index_migration_to_v2.ipynb b/docs/examples_notebooks/index_migration_to_v2.ipynb
@@ -70,7 +70,6 @@
    "outputs": [],
    "source": [
     "import numpy as np\n",
-    "\n",
     "from graphrag.utils.storage import (\n",
     "    delete_table_from_storage,\n",
     "    load_table_from_storage,\n",
diff --git a/docs/examples_notebooks/input_documents.ipynb b/docs/examples_notebooks/input_documents.ipynb
@@ -30,9 +30,8 @@
     "from pathlib import Path\n",
     "from pprint import pprint\n",
     "\n",
-    "import pandas as pd\n",
-    "\n",
     "import graphrag.api as api\n",
+    "import pandas as pd\n",
     "from graphrag.config.load_config import load_config\n",
     "from graphrag.index.typing.pipeline_run_result import PipelineRunResult"
    ]
diff --git a/docs/examples_notebooks/local_search.ipynb b/docs/examples_notebooks/local_search.ipynb
@@ -19,7 +19,6 @@
     "import os\n",
     "\n",
     "import pandas as pd\n",
-    "\n",
     "from graphrag.config.models.vector_store_schema_config import VectorStoreSchemaConfig\n",
     "from graphrag.query.context_builder.entity_extraction import EntityVectorStoreKey\n",
     "from graphrag.query.indexer_adapters import (\n",
diff --git a/tests/__init__.py b/tests/__init__.py
@@ -7,6 +7,7 @@
 # Register MOCK providers
 from graphrag.config.enums import ModelType
 from graphrag.language_model.factory import ChatModelFactory, EmbeddingModelFactory
+
 from tests.mock_provider import MockChatLLM, MockEmbeddingLLM
 
 ChatModelFactory().register(ModelType.MockChat, lambda **kwargs: MockChatLLM(**kwargs))
diff --git a/tests/integration/cache/test_factory.py b/tests/integration/cache/test_factory.py
@@ -8,7 +8,6 @@
 import sys
 
 import pytest
-
 from graphrag.cache.factory import CacheFactory
 from graphrag.cache.json_pipeline_cache import JsonPipelineCache
 from graphrag.cache.memory_pipeline_cache import InMemoryCache
diff --git a/tests/integration/language_model/test_rate_limiter.py b/tests/integration/language_model/test_rate_limiter.py
@@ -9,13 +9,13 @@
 from queue import Queue
 
 import pytest
-
 from graphrag.language_model.providers.litellm.services.rate_limiter.rate_limiter import (
     RateLimiter,
 )
 from graphrag.language_model.providers.litellm.services.rate_limiter.rate_limiter_factory import (
     RateLimiterFactory,
 )
+
 from tests.integration.language_model.utils import (
     assert_max_num_values_per_period,
     assert_stagger,
diff --git a/tests/integration/language_model/test_retries.py b/tests/integration/language_model/test_retries.py
@@ -6,7 +6,6 @@
 import time
 
 import pytest
-
 from graphrag.language_model.providers.litellm.services.retry.retry_factory import (
     RetryFactory,
 )
diff --git a/tests/integration/logging/test_factory.py b/tests/integration/logging/test_factory.py
@@ -8,7 +8,6 @@
 import logging
 
 import pytest
-
 from graphrag.config.enums import ReportingType
 from graphrag.logger.blob_workflow_logger import BlobWorkflowLogger
 from graphrag.logger.factory import LoggerFactory
diff --git a/tests/integration/logging/test_standard_logging.py b/tests/integration/logging/test_standard_logging.py
@@ -8,6 +8,7 @@
 from pathlib import Path
 
 from graphrag.logger.standard_logging import DEFAULT_LOG_FILENAME, init_loggers
+
 from tests.unit.config.utils import get_default_graphrag_config
 
 
diff --git a/tests/integration/storage/test_cosmosdb_storage.py b/tests/integration/storage/test_cosmosdb_storage.py
@@ -8,7 +8,6 @@
 from datetime import datetime
 
 import pytest
-
 from graphrag.storage.cosmosdb_pipeline_storage import CosmosDBPipelineStorage
 
 # cspell:disable-next-line well-known-key
diff --git a/tests/integration/storage/test_factory.py b/tests/integration/storage/test_factory.py
@@ -8,7 +8,6 @@
 import sys
 
 import pytest
-
 from graphrag.config.enums import StorageType
 from graphrag.storage.blob_pipeline_storage import BlobPipelineStorage
 from graphrag.storage.cosmosdb_pipeline_storage import CosmosDBPipelineStorage
diff --git a/tests/integration/vector_stores/test_azure_ai_search.py b/tests/integration/vector_stores/test_azure_ai_search.py
@@ -7,7 +7,6 @@
 from unittest.mock import MagicMock, patch
 
 import pytest
-
 from graphrag.config.models.vector_store_schema_config import VectorStoreSchemaConfig
 from graphrag.vector_stores.azure_ai_search import AzureAISearchVectorStore
 from graphrag.vector_stores.base import VectorStoreDocument
diff --git a/tests/integration/vector_stores/test_cosmosdb.py b/tests/integration/vector_stores/test_cosmosdb.py
@@ -7,7 +7,6 @@
 
 import numpy as np
 import pytest
-
 from graphrag.config.models.vector_store_schema_config import VectorStoreSchemaConfig
 from graphrag.vector_stores.base import VectorStoreDocument
 from graphrag.vector_stores.cosmosdb import CosmosDBVectorStore
diff --git a/tests/integration/vector_stores/test_factory.py b/tests/integration/vector_stores/test_factory.py
@@ -6,7 +6,6 @@
 """
 
 import pytest
-
 from graphrag.config.enums import VectorStoreType
 from graphrag.config.models.vector_store_schema_config import VectorStoreSchemaConfig
 from graphrag.vector_stores.azure_ai_search import AzureAISearchVectorStore
diff --git a/tests/integration/vector_stores/test_lancedb.py b/tests/integration/vector_stores/test_lancedb.py
@@ -8,7 +8,6 @@
 
 import numpy as np
 import pytest
-
 from graphrag.config.models.vector_store_schema_config import VectorStoreSchemaConfig
 from graphrag.vector_stores.base import VectorStoreDocument
 from graphrag.vector_stores.lancedb import LanceDBVectorStore
diff --git a/tests/mock_provider.py b/tests/mock_provider.py
@@ -6,15 +6,14 @@
 from collections.abc import AsyncGenerator, Generator
 from typing import Any
 
-from pydantic import BaseModel
-
 from graphrag.config.enums import ModelType
 from graphrag.config.models.language_model_config import LanguageModelConfig
 from graphrag.language_model.response.base import (
     BaseModelOutput,
     BaseModelResponse,
     ModelResponse,
 )
+from pydantic import BaseModel
 
 
 class MockChatLLM:
diff --git a/tests/smoke/test_fixtures.py b/tests/smoke/test_fixtures.py
@@ -14,7 +14,6 @@
 
 import pandas as pd
 import pytest
-
 from graphrag.query.context_builder.community_context import (
     NO_COMMUNITY_RECORDS_WARNING,
 )
diff --git a/tests/unit/config/test_config.py b/tests/unit/config/test_config.py
@@ -5,13 +5,13 @@
 from pathlib import Path
 from unittest import mock
 
-import pytest
-from pydantic import ValidationError
-
 import graphrag.config.defaults as defs
+import pytest
 from graphrag.config.create_graphrag_config import create_graphrag_config
 from graphrag.config.enums import AuthType, ModelType
 from graphrag.config.load_config import load_config
+from pydantic import ValidationError
+
 from tests.unit.config.utils import (
     DEFAULT_EMBEDDING_MODEL_CONFIG,
     DEFAULT_MODEL_CONFIG,
diff --git a/tests/unit/config/utils.py b/tests/unit/config/utils.py
@@ -3,8 +3,6 @@
 
 from dataclasses import asdict
 
-from pydantic import BaseModel
-
 import graphrag.config.defaults as defs
 from graphrag.config.models.basic_search_config import BasicSearchConfig
 from graphrag.config.models.cache_config import CacheConfig
@@ -32,6 +30,7 @@
     SummarizeDescriptionsConfig,
 )
 from graphrag.config.models.vector_store_config import VectorStoreConfig
+from pydantic import BaseModel
 
 FAKE_API_KEY = "NOT_AN_API_KEY"
 
diff --git a/tests/unit/indexing/graph/utils/test_stable_lcc.py b/tests/unit/indexing/graph/utils/test_stable_lcc.py
@@ -3,7 +3,6 @@
 import unittest
 
 import networkx as nx
-
 from graphrag.index.utils.stable_lcc import stable_largest_connected_component
 
 
diff --git a/tests/unit/indexing/operations/chunk_text/test_chunk_text.py b/tests/unit/indexing/operations/chunk_text/test_chunk_text.py
@@ -7,7 +7,6 @@
 
 import pandas as pd
 import pytest
-
 from graphrag.config.enums import ChunkStrategyType
 from graphrag.index.operations.chunk_text.chunk_text import (
     _get_num_total,
diff --git a/tests/unit/indexing/test_init_content.py b/tests/unit/indexing/test_init_content.py
@@ -5,7 +5,6 @@
 from typing import Any, cast
 
 import yaml
-
 from graphrag.config.create_graphrag_config import create_graphrag_config
 from graphrag.config.init_content import INIT_YAML
 from graphrag.config.models.graph_rag_config import GraphRagConfig
diff --git a/tests/unit/indexing/text_splitting/test_text_splitting.py b/tests/unit/indexing/text_splitting/test_text_splitting.py
@@ -6,7 +6,6 @@
 
 import pytest
 import tiktoken
-
 from graphrag.index.text_splitting.text_splitting import (
     NoopTextSplitter,
     TokenChunkerOptions,
diff --git a/tests/unit/indexing/verbs/entities/extraction/strategies/graph_intelligence/test_gi_entity_extraction.py b/tests/unit/indexing/verbs/entities/extraction/strategies/graph_intelligence/test_gi_entity_extraction.py
@@ -7,6 +7,7 @@
     Document,
 )
 from graphrag.prompts.index.extract_graph import GRAPH_EXTRACTION_PROMPT
+
 from tests.unit.indexing.verbs.helpers.mock_llm import create_mock_llm
 
 
diff --git a/tests/unit/indexing/verbs/helpers/mock_llm.py b/tests/unit/indexing/verbs/helpers/mock_llm.py
@@ -1,9 +1,8 @@
 # Copyright (c) 2024 Microsoft Corporation.
 # Licensed under the MIT License
-from pydantic import BaseModel
-
 from graphrag.language_model.manager import ModelManager
 from graphrag.language_model.protocol.base import ChatModel
+from pydantic import BaseModel
 
 
 def create_mock_llm(responses: list[str | BaseModel], name: str = "mock") -> ChatModel:
diff --git a/tests/unit/utils/test_embeddings.py b/tests/unit/utils/test_embeddings.py
@@ -2,7 +2,6 @@
 # Licensed under the MIT License
 
 import pytest
-
 from graphrag.config.embeddings import create_index_name
 
 
diff --git a/tests/verbs/test_extract_covariates.py b/tests/verbs/test_extract_covariates.py
@@ -1,15 +1,14 @@
 # Copyright (c) 2024 Microsoft Corporation.
 # Licensed under the MIT License
 
-from pandas.testing import assert_series_equal
-
 from graphrag.config.create_graphrag_config import create_graphrag_config
 from graphrag.config.enums import ModelType
 from graphrag.data_model.schemas import COVARIATES_FINAL_COLUMNS
 from graphrag.index.workflows.extract_covariates import (
     run_workflow,
 )
 from graphrag.utils.storage import load_table_from_storage
+from pandas.testing import assert_series_equal
 
 from .util import (
     DEFAULT_MODEL_CONFIG,
diff --git a/tests/verbs/test_pipeline_state.py b/tests/verbs/test_pipeline_state.py
@@ -9,6 +9,7 @@
 from graphrag.index.typing.context import PipelineRunContext
 from graphrag.index.typing.workflow import WorkflowFunctionOutput
 from graphrag.index.workflows.factory import PipelineFactory
+
 from tests.verbs.util import DEFAULT_MODEL_CONFIG
 
 
diff --git a/tests/verbs/util.py b/tests/verbs/util.py
@@ -1,13 +1,12 @@
 # Copyright (c) 2024 Microsoft Corporation.
 # Licensed under the MIT License
 
-import pandas as pd
-from pandas.testing import assert_series_equal
-
 import graphrag.config.defaults as defs
+import pandas as pd
 from graphrag.index.run.utils import create_run_context
 from graphrag.index.typing.context import PipelineRunContext
 from graphrag.utils.storage import load_table_from_storage, write_table_to_storage
+from pandas.testing import assert_series_equal
 
 pd.set_option("display.max_columns", None)
 
diff --git a/unified-search-app/app/app_logic.py b/unified-search-app/app/app_logic.py
@@ -7,6 +7,7 @@
 import logging
 from typing import TYPE_CHECKING
 
+import graphrag.api as api
 import streamlit as st
 from knowledge_loader.data_sources.loader import (
     create_datasource,
@@ -17,8 +18,6 @@
 from state.session_variables import SessionVariables
 from ui.search import display_search_result
 
-import graphrag.api as api
-
 if TYPE_CHECKING:
     import pandas as pd
 
diff --git a/unified-search-app/app/knowledge_loader/data_sources/blob_source.py b/unified-search-app/app/knowledge_loader/data_sources/blob_source.py
@@ -13,10 +13,9 @@
 import yaml
 from azure.identity import DefaultAzureCredential
 from azure.storage.blob import BlobServiceClient, ContainerClient
-from knowledge_loader.data_sources.typing import Datasource
-
 from graphrag.config.create_graphrag_config import create_graphrag_config
 from graphrag.config.models.graph_rag_config import GraphRagConfig
+from knowledge_loader.data_sources.typing import Datasource
 
 from .default import blob_account_name, blob_container_name
 
diff --git a/unified-search-app/app/knowledge_loader/data_sources/local_source.py b/unified-search-app/app/knowledge_loader/data_sources/local_source.py
@@ -8,10 +8,9 @@
 from pathlib import Path
 
 import pandas as pd
-from knowledge_loader.data_sources.typing import Datasource
-
 from graphrag.config.load_config import load_config
 from graphrag.config.models.graph_rag_config import GraphRagConfig
+from knowledge_loader.data_sources.typing import Datasource
 
 logging.basicConfig(level=logging.INFO)
 logging.getLogger("azure").setLevel(logging.WARNING)
diff --git a/unified-search-app/app/knowledge_loader/data_sources/typing.py b/unified-search-app/app/knowledge_loader/data_sources/typing.py
@@ -8,7 +8,6 @@
 from enum import Enum
 
 import pandas as pd
-
 from graphrag.config.models.graph_rag_config import GraphRagConfig
 
 
