reformat code

miguelgrinberg · miguelgrinberg · commit 5dd57f6e01d1 · 2025-01-13T15:59:21.000Z
diff --git a/elasticsearch/dsl/_async/document.py b/elasticsearch/dsl/_async/document.py
@@ -28,9 +28,10 @@
     cast,
 )
 
+from typing_extensions import Self, dataclass_transform
+
 from elasticsearch.exceptions import NotFoundError, RequestError
 from elasticsearch.helpers import async_bulk
-from typing_extensions import Self, dataclass_transform
 
 from .._async.index import AsyncIndex
 from ..async_connections import get_connection
diff --git a/elasticsearch/dsl/_async/faceted_search.py b/elasticsearch/dsl/_async/faceted_search.py
@@ -18,7 +18,6 @@
 from typing import TYPE_CHECKING
 
 from ..faceted_search_base import FacetedResponse, FacetedSearchBase
-
 from ..utils import _R
 from .search import AsyncSearch
 
diff --git a/elasticsearch/dsl/_async/index.py b/elasticsearch/dsl/_async/index.py
@@ -29,6 +29,7 @@
 
 if TYPE_CHECKING:
     from elastic_transport import ObjectApiResponse
+
     from elasticsearch import AsyncElasticsearch
 
 
diff --git a/elasticsearch/dsl/_async/search.py b/elasticsearch/dsl/_async/search.py
@@ -27,9 +27,10 @@
     cast,
 )
 
+from typing_extensions import Self
+
 from elasticsearch.exceptions import ApiError
 from elasticsearch.helpers import async_scan
-from typing_extensions import Self
 
 from ..async_connections import get_connection
 from ..response import Response
diff --git a/elasticsearch/dsl/_sync/document.py b/elasticsearch/dsl/_sync/document.py
@@ -28,9 +28,10 @@
     cast,
 )
 
+from typing_extensions import Self, dataclass_transform
+
 from elasticsearch.exceptions import NotFoundError, RequestError
 from elasticsearch.helpers import bulk
-from typing_extensions import Self, dataclass_transform
 
 from .._sync.index import Index
 from ..connections import get_connection
diff --git a/elasticsearch/dsl/_sync/faceted_search.py b/elasticsearch/dsl/_sync/faceted_search.py
@@ -18,7 +18,6 @@
 from typing import TYPE_CHECKING
 
 from ..faceted_search_base import FacetedResponse, FacetedSearchBase
-
 from ..utils import _R
 from .search import Search
 
diff --git a/elasticsearch/dsl/_sync/index.py b/elasticsearch/dsl/_sync/index.py
@@ -29,6 +29,7 @@
 
 if TYPE_CHECKING:
     from elastic_transport import ObjectApiResponse
+
     from elasticsearch import Elasticsearch
 
 
diff --git a/elasticsearch/dsl/_sync/search.py b/elasticsearch/dsl/_sync/search.py
@@ -18,9 +18,10 @@
 import contextlib
 from typing import TYPE_CHECKING, Any, Dict, Iterator, List, Optional, cast
 
+from typing_extensions import Self
+
 from elasticsearch.exceptions import ApiError
 from elasticsearch.helpers import scan
-from typing_extensions import Self
 
 from ..connections import get_connection
 from ..response import Response
diff --git a/elasticsearch/dsl/aggs.py b/elasticsearch/dsl/aggs.py
@@ -43,7 +43,6 @@
     from elastic_transport.client_utils import DefaultType
 
     from . import types
-
     from .document_base import InstrumentedField
     from .search_base import SearchBase
 
diff --git a/elasticsearch/dsl/query.py b/elasticsearch/dsl/query.py
@@ -49,7 +49,6 @@
     from elastic_transport.client_utils import DefaultType
 
     from . import types, wrappers
-
     from .document_base import InstrumentedField
 
 _T = TypeVar("_T")
diff --git a/elasticsearch/dsl/search.py b/elasticsearch/dsl/search.py
@@ -15,14 +15,6 @@
 #  specific language governing permissions and limitations
 #  under the License.
 
-from ._async.search import (  # noqa: F401
-    AsyncEmptySearch,
-    AsyncMultiSearch,
-    AsyncSearch,
-)
-from ._sync.search import (  # noqa: F401
-    EmptySearch,
-    MultiSearch,
-    Search,
-)
+from ._async.search import AsyncEmptySearch, AsyncMultiSearch, AsyncSearch  # noqa: F401
+from ._sync.search import EmptySearch, MultiSearch, Search  # noqa: F401
 from .search_base import Q  # noqa: F401
diff --git a/elasticsearch/dsl/utils.py b/elasticsearch/dsl/utils.py
@@ -43,6 +43,7 @@
 
 if TYPE_CHECKING:
     from elastic_transport import ObjectApiResponse
+
     from elasticsearch import AsyncElasticsearch, Elasticsearch
 
     from .document_base import DocumentOptions
diff --git a/examples/dsl/async/composite_agg.py b/examples/dsl/async/composite_agg.py
@@ -19,10 +19,9 @@
 import os
 from typing import Any, AsyncIterator, Dict, Mapping, Sequence, cast
 
-from elasticsearch.helpers import async_bulk
-
 from elasticsearch.dsl import Agg, AsyncSearch, Response, aggs, async_connections
 from elasticsearch.dsl.types import CompositeAggregate
+from elasticsearch.helpers import async_bulk
 from test_elasticsearch.test_dsl.test_integration.test_data import DATA, GIT_INDEX
 
 
diff --git a/examples/dsl/composite_agg.py b/examples/dsl/composite_agg.py
@@ -18,10 +18,9 @@
 import os
 from typing import Any, Dict, Iterator, Mapping, Sequence, cast
 
-from elasticsearch.helpers import bulk
-
 from elasticsearch.dsl import Agg, Response, Search, aggs, connections
 from elasticsearch.dsl.types import CompositeAggregate
+from elasticsearch.helpers import bulk
 from test_elasticsearch.test_dsl.test_integration.test_data import DATA, GIT_INDEX
 
 
diff --git a/setup.cfg b/setup.cfg
@@ -1,2 +1,2 @@
 [flake8]
-ignore = E203, E266, E501, W503
+ignore = E203, E266, E501, W503, E704, E741
diff --git a/test_elasticsearch/test_dsl/conftest.py b/test_elasticsearch/test_dsl/conftest.py
@@ -27,15 +27,15 @@
 
 import pytest_asyncio
 from elastic_transport import ObjectApiResponse
-from elasticsearch import AsyncElasticsearch, Elasticsearch
-from elasticsearch.exceptions import ConnectionError
-from elasticsearch.helpers import bulk
 from pytest import fixture, skip
 
+from elasticsearch import AsyncElasticsearch, Elasticsearch
 from elasticsearch.dsl import Search
 from elasticsearch.dsl.async_connections import add_connection as add_async_connection
 from elasticsearch.dsl.async_connections import connections as async_connections
 from elasticsearch.dsl.connections import add_connection, connections
+from elasticsearch.exceptions import ConnectionError
+from elasticsearch.helpers import bulk
 
 from .test_integration._async import test_document as async_document
 from .test_integration._sync import test_document as sync_document
diff --git a/test_elasticsearch/test_dsl/test_connections.py b/test_elasticsearch/test_dsl/test_connections.py
@@ -17,9 +17,9 @@
 
 from typing import Any, List
 
-from elasticsearch import Elasticsearch
 from pytest import raises
 
+from elasticsearch import Elasticsearch
 from elasticsearch.dsl import connections, serializer
 
 
diff --git a/test_elasticsearch/test_dsl/test_integration/_async/test_analysis.py b/test_elasticsearch/test_dsl/test_integration/_async/test_analysis.py
@@ -16,8 +16,8 @@
 #  under the License.
 
 import pytest
-from elasticsearch import AsyncElasticsearch
 
+from elasticsearch import AsyncElasticsearch
 from elasticsearch.dsl import analyzer, token_filter, tokenizer
 
 
diff --git a/test_elasticsearch/test_dsl/test_integration/_async/test_document.py b/test_elasticsearch/test_dsl/test_integration/_async/test_document.py
@@ -26,11 +26,10 @@
 from typing import TYPE_CHECKING, Any, AsyncIterator, Dict, List, Tuple, Union
 
 import pytest
-from elasticsearch import AsyncElasticsearch, ConflictError, NotFoundError
-from elasticsearch.helpers.errors import BulkIndexError
 from pytest import raises
 from pytz import timezone
 
+from elasticsearch import AsyncElasticsearch, ConflictError, NotFoundError
 from elasticsearch.dsl import (
     AsyncDocument,
     AsyncSearch,
@@ -54,6 +53,7 @@
     mapped_field,
 )
 from elasticsearch.dsl.utils import AttrList
+from elasticsearch.helpers.errors import BulkIndexError
 
 snowball = analyzer("my_snow", tokenizer="standard", filter=["lowercase", "snowball"])
 
diff --git a/test_elasticsearch/test_dsl/test_integration/_async/test_faceted_search.py b/test_elasticsearch/test_dsl/test_integration/_async/test_faceted_search.py
@@ -19,8 +19,8 @@
 from typing import Tuple, Type
 
 import pytest
-from elasticsearch import AsyncElasticsearch
 
+from elasticsearch import AsyncElasticsearch
 from elasticsearch.dsl import A, AsyncDocument, AsyncSearch, Boolean, Date, Keyword
 from elasticsearch.dsl.faceted_search import (
     AsyncFacetedSearch,
diff --git a/test_elasticsearch/test_dsl/test_integration/_async/test_index.py b/test_elasticsearch/test_dsl/test_integration/_async/test_index.py
@@ -16,8 +16,8 @@
 #  under the License.
 
 import pytest
-from elasticsearch import AsyncElasticsearch
 
+from elasticsearch import AsyncElasticsearch
 from elasticsearch.dsl import (
     AsyncComposableIndexTemplate,
     AsyncDocument,
diff --git a/test_elasticsearch/test_dsl/test_integration/_async/test_mapping.py b/test_elasticsearch/test_dsl/test_integration/_async/test_mapping.py
@@ -16,9 +16,9 @@
 #  under the License.
 
 import pytest
-from elasticsearch import AsyncElasticsearch
 from pytest import raises
 
+from elasticsearch import AsyncElasticsearch
 from elasticsearch.dsl import AsyncMapping, analysis, exceptions
 
 
diff --git a/test_elasticsearch/test_dsl/test_integration/_async/test_search.py b/test_elasticsearch/test_dsl/test_integration/_async/test_search.py
@@ -17,9 +17,9 @@
 
 
 import pytest
-from elasticsearch import ApiError, AsyncElasticsearch
 from pytest import raises
 
+from elasticsearch import ApiError, AsyncElasticsearch
 from elasticsearch.dsl import (
     AsyncDocument,
     AsyncMultiSearch,
diff --git a/test_elasticsearch/test_dsl/test_integration/_async/test_update_by_query.py b/test_elasticsearch/test_dsl/test_integration/_async/test_update_by_query.py
@@ -16,8 +16,8 @@
 #  under the License.
 
 import pytest
-from elasticsearch import AsyncElasticsearch
 
+from elasticsearch import AsyncElasticsearch
 from elasticsearch.dsl import AsyncUpdateByQuery
 from elasticsearch.dsl.search import Q
 
diff --git a/test_elasticsearch/test_dsl/test_integration/_sync/test_analysis.py b/test_elasticsearch/test_dsl/test_integration/_sync/test_analysis.py
@@ -16,8 +16,8 @@
 #  under the License.
 
 import pytest
-from elasticsearch import Elasticsearch
 
+from elasticsearch import Elasticsearch
 from elasticsearch.dsl import analyzer, token_filter, tokenizer
 
 
diff --git a/test_elasticsearch/test_dsl/test_integration/_sync/test_document.py b/test_elasticsearch/test_dsl/test_integration/_sync/test_document.py
@@ -26,11 +26,10 @@
 from typing import TYPE_CHECKING, Any, Dict, Iterator, List, Tuple, Union
 
 import pytest
-from elasticsearch import ConflictError, Elasticsearch, NotFoundError
-from elasticsearch.helpers.errors import BulkIndexError
 from pytest import raises
 from pytz import timezone
 
+from elasticsearch import ConflictError, Elasticsearch, NotFoundError
 from elasticsearch.dsl import (
     Binary,
     Boolean,
@@ -54,6 +53,7 @@
     mapped_field,
 )
 from elasticsearch.dsl.utils import AttrList
+from elasticsearch.helpers.errors import BulkIndexError
 
 snowball = analyzer("my_snow", tokenizer="standard", filter=["lowercase", "snowball"])
 
diff --git a/test_elasticsearch/test_dsl/test_integration/_sync/test_faceted_search.py b/test_elasticsearch/test_dsl/test_integration/_sync/test_faceted_search.py
@@ -19,8 +19,8 @@
 from typing import Tuple, Type
 
 import pytest
-from elasticsearch import Elasticsearch
 
+from elasticsearch import Elasticsearch
 from elasticsearch.dsl import A, Boolean, Date, Document, Keyword, Search
 from elasticsearch.dsl.faceted_search import (
     DateHistogramFacet,
diff --git a/test_elasticsearch/test_dsl/test_integration/_sync/test_index.py b/test_elasticsearch/test_dsl/test_integration/_sync/test_index.py
@@ -16,8 +16,8 @@
 #  under the License.
 
 import pytest
-from elasticsearch import Elasticsearch
 
+from elasticsearch import Elasticsearch
 from elasticsearch.dsl import (
     ComposableIndexTemplate,
     Date,
diff --git a/test_elasticsearch/test_dsl/test_integration/_sync/test_mapping.py b/test_elasticsearch/test_dsl/test_integration/_sync/test_mapping.py
@@ -16,9 +16,9 @@
 #  under the License.
 
 import pytest
-from elasticsearch import Elasticsearch
 from pytest import raises
 
+from elasticsearch import Elasticsearch
 from elasticsearch.dsl import Mapping, analysis, exceptions
 
 
diff --git a/test_elasticsearch/test_dsl/test_integration/_sync/test_search.py b/test_elasticsearch/test_dsl/test_integration/_sync/test_search.py
@@ -17,9 +17,9 @@
 
 
 import pytest
-from elasticsearch import ApiError, Elasticsearch
 from pytest import raises
 
+from elasticsearch import ApiError, Elasticsearch
 from elasticsearch.dsl import Date, Document, Keyword, MultiSearch, Q, Search, Text
 from elasticsearch.dsl.response import aggs
 
diff --git a/test_elasticsearch/test_dsl/test_integration/_sync/test_update_by_query.py b/test_elasticsearch/test_dsl/test_integration/_sync/test_update_by_query.py
@@ -16,8 +16,8 @@
 #  under the License.
 
 import pytest
-from elasticsearch import Elasticsearch
 
+from elasticsearch import Elasticsearch
 from elasticsearch.dsl import UpdateByQuery
 from elasticsearch.dsl.search import Q
 
diff --git a/test_elasticsearch/test_dsl/test_integration/test_count.py b/test_elasticsearch/test_dsl/test_integration/test_count.py
@@ -18,7 +18,6 @@
 from typing import Any
 
 from elasticsearch import Elasticsearch
-
 from elasticsearch.dsl.search import Q, Search
 
 
diff --git a/test_elasticsearch/test_dsl/test_integration/test_examples/_async/test_alias_migration.py b/test_elasticsearch/test_dsl/test_integration/test_examples/_async/test_alias_migration.py
@@ -16,6 +16,7 @@
 #  under the License.
 
 import pytest
+
 from elasticsearch import AsyncElasticsearch
 
 from ..async_examples import alias_migration
diff --git a/test_elasticsearch/test_dsl/test_integration/test_examples/_async/test_completion.py b/test_elasticsearch/test_dsl/test_integration/test_examples/_async/test_completion.py
@@ -16,6 +16,7 @@
 #  under the License.
 
 import pytest
+
 from elasticsearch import AsyncElasticsearch
 
 from ..async_examples.completion import Person
diff --git a/test_elasticsearch/test_dsl/test_integration/test_examples/_async/test_composite_aggs.py b/test_elasticsearch/test_dsl/test_integration/test_examples/_async/test_composite_aggs.py
@@ -16,8 +16,8 @@
 #  under the License.
 
 import pytest
-from elasticsearch import AsyncElasticsearch
 
+from elasticsearch import AsyncElasticsearch
 from elasticsearch.dsl import A, AsyncSearch
 
 from ..async_examples.composite_agg import scan_aggs
diff --git a/test_elasticsearch/test_dsl/test_integration/test_examples/_async/test_parent_child.py b/test_elasticsearch/test_dsl/test_integration/test_examples/_async/test_parent_child.py
@@ -19,8 +19,8 @@
 
 import pytest
 import pytest_asyncio
-from elasticsearch import AsyncElasticsearch
 
+from elasticsearch import AsyncElasticsearch
 from elasticsearch.dsl import Q
 
 from ..async_examples.parent_child import Answer, Comment, Question, User, setup
diff --git a/test_elasticsearch/test_dsl/test_integration/test_examples/_async/test_percolate.py b/test_elasticsearch/test_dsl/test_integration/test_examples/_async/test_percolate.py
@@ -16,6 +16,7 @@
 #  under the License.
 
 import pytest
+
 from elasticsearch import AsyncElasticsearch
 
 from ..async_examples.percolate import BlogPost, setup
diff --git a/test_elasticsearch/test_dsl/test_integration/test_examples/_async/test_vectors.py b/test_elasticsearch/test_dsl/test_integration/test_examples/_async/test_vectors.py
diff --git a/test_elasticsearch/test_dsl/test_integration/test_examples/_sync/test_alias_migration.py b/test_elasticsearch/test_dsl/test_integration/test_examples/_sync/test_alias_migration.py
diff --git a/test_elasticsearch/test_dsl/test_integration/test_examples/_sync/test_completion.py b/test_elasticsearch/test_dsl/test_integration/test_examples/_sync/test_completion.py
diff --git a/test_elasticsearch/test_dsl/test_integration/test_examples/_sync/test_composite_aggs.py b/test_elasticsearch/test_dsl/test_integration/test_examples/_sync/test_composite_aggs.py
diff --git a/test_elasticsearch/test_dsl/test_integration/test_examples/_sync/test_parent_child.py b/test_elasticsearch/test_dsl/test_integration/test_examples/_sync/test_parent_child.py
diff --git a/test_elasticsearch/test_dsl/test_integration/test_examples/_sync/test_percolate.py b/test_elasticsearch/test_dsl/test_integration/test_examples/_sync/test_percolate.py
diff --git a/test_elasticsearch/test_dsl/test_integration/test_examples/_sync/test_vectors.py b/test_elasticsearch/test_dsl/test_integration/test_examples/_sync/test_vectors.py
diff --git a/test_elasticsearch/test_server/test_vectorstore/test_vectorstore.py b/test_elasticsearch/test_server/test_vectorstore/test_vectorstore.py
