chore: add annotations future import to all non-test files

Author: cpcloud

ci/make_geography_db.py

@@ -13,6 +13,8 @@
 #
 # The source of the `independence` table has been obtained from
 # [Wikipedia](https://en.wikipedia.org/wiki/List_of_national_independence_days).
+from __future__ import annotations
+
 import argparse
 import datetime
 import tempfile

docs/example.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import ibis
 
 con = ibis.sqlite.connect("crunchbase.db")

docs/sqlalchemy_example.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import sqlalchemy as sa
 
 c = sa.table("companies")

gen_matrix.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from pathlib import Path
 
 import pandas as pd

ibis/backends/base/sql/alchemy/geospatial.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 try:
     import geoalchemy2
     import geoalchemy2.shape  # noqa: F401

ibis/backends/base/sql/alchemy/registry.py

@@ -1,6 +1,8 @@
+from __future__ import annotations
+
 import functools
 import operator
-from typing import Any, Dict
+from typing import Any
 
 import sqlalchemy as sa
 
@@ -482,7 +484,7 @@ def translator(t, op: ops.Node):
     return translator
 
 
-sqlalchemy_operation_registry: Dict[Any, Any] = {
+sqlalchemy_operation_registry: dict[Any, Any] = {
     ops.Alias: _alias,
     ops.And: fixed_arity(operator.and_, 2),
     ops.Or: fixed_arity(operator.or_, 2),

ibis/backends/base/sql/compiler/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from ibis.backends.base.sql.compiler.base import DDL, DML
 from ibis.backends.base.sql.compiler.query_builder import (
     Compiler,

ibis/backends/base/sql/compiler/base.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import abc
 from itertools import chain
 

ibis/backends/base/sql/ddl.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import re
 
 import ibis.expr.datatypes as dt

ibis/backends/base/sql/registry/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from ibis.backends.base.sql.registry.aggregate import reduction
 from ibis.backends.base.sql.registry.helpers import (
     quote_identifier,

ibis/backends/base/sql/registry/aggregate.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import itertools
 
 import ibis

ibis/backends/base/sql/registry/case.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from io import StringIO
 
 

ibis/backends/base/sql/registry/helpers.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import ibis.common.exceptions as com
 import ibis.expr.operations as ops
 from ibis.backends.base.sql.registry import identifiers

ibis/backends/base/sql/registry/identifiers.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 # Copyright 2014 Cloudera Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");

ibis/backends/base/sql/registry/literal.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import datetime
 import math
 

ibis/backends/base/sql/registry/main.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import ibis.common.exceptions as com
 import ibis.expr.datatypes as dt
 import ibis.expr.operations as ops

ibis/backends/base/sql/registry/string.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import ibis.expr.operations as ops
 from ibis.backends.base.sql.registry import helpers
 

ibis/backends/base/sql/registry/timestamp.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import ibis.common.exceptions as com
 import ibis.expr.datatypes as dt
 import ibis.expr.operations as ops

ibis/backends/bigquery/__init__.py

@@ -1,6 +1,8 @@
 """BigQuery public API."""
+
+from __future__ import annotations
+
 import warnings
-from typing import Optional, Tuple
 from urllib.parse import parse_qs, urlparse
 
 import google.auth.credentials
@@ -73,14 +75,14 @@ def _from_url(self, url):
 
     def connect(
         self,
-        project_id: Optional[str] = None,
+        project_id: str | None = None,
         dataset_id: str = "",
-        credentials: Optional[google.auth.credentials.Credentials] = None,
-        application_name: Optional[str] = None,
+        credentials: google.auth.credentials.Credentials | None = None,
+        application_name: str | None = None,
         auth_local_webserver: bool = True,
         auth_external_data: bool = False,
         auth_cache: str = "default",
-        partition_column: Optional[str] = "PARTITIONTIME",
+        partition_column: str | None = "PARTITIONTIME",
     ) -> "Backend":
         """Create a :class:`Backend` for use with Ibis.
 
@@ -180,7 +182,7 @@ def connect(
 
         return new_backend
 
-    def _parse_project_and_dataset(self, dataset) -> Tuple[str, str]:
+    def _parse_project_and_dataset(self, dataset) -> tuple[str, str]:
         if not dataset and not self.dataset:
             raise ValueError("Unable to determine BigQuery dataset.")
         project, _, dataset = parse_project_and_dataset(
@@ -406,14 +408,14 @@ def compile(expr, params=None, **kwargs):
 
 
 def connect(
-    project_id: Optional[str] = None,
+    project_id: str | None = None,
     dataset_id: str = "",
-    credentials: Optional[google.auth.credentials.Credentials] = None,
-    application_name: Optional[str] = None,
+    credentials: google.auth.credentials.Credentials | None = None,
+    application_name: str | None = None,
     auth_local_webserver: bool = False,
     auth_external_data: bool = False,
     auth_cache: str = "default",
-    partition_column: Optional[str] = "PARTITIONTIME",
+    partition_column: str | None = "PARTITIONTIME",
 ) -> Backend:
     """Create a :class:`Backend` for use with Ibis.
 

ibis/backends/bigquery/compiler.py

@@ -1,5 +1,7 @@
 """Module to convert from Ibis expression to SQL string."""
 
+from __future__ import annotations
+
 import re
 from functools import partial
 

ibis/backends/bigquery/datatypes.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from multipledispatch import Dispatcher
 
 import ibis.expr.datatypes as dt

ibis/backends/bigquery/operations.py

@@ -1,5 +1,7 @@
 """Ibis operations specific to BigQuery."""
 
+from __future__ import annotations
+
 import ibis.expr.operations as ops
 
 

ibis/backends/bigquery/registry.py

@@ -1,5 +1,7 @@
 """Module to convert from Ibis expression to SQL string."""
 
+from __future__ import annotations
+
 import base64
 import datetime
 from typing import Literal

ibis/backends/bigquery/rewrites.py

@@ -1,5 +1,7 @@
 """Methods to translate BigQuery expressions before compilation."""
 
+from __future__ import annotations
+
 import toolz
 
 import ibis.expr.datatypes as dt

ibis/backends/bigquery/tests/conftest.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import concurrent.futures
 import functools
 import io

ibis/backends/bigquery/tests/system/conftest.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import os
 
 import google.auth

ibis/backends/bigquery/udf/core.py

@@ -1,5 +1,7 @@
 """Translate a Python AST to JavaScript."""
 
+from __future__ import annotations
+
 import ast
 import contextlib
 import functools

ibis/backends/bigquery/udf/find.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import ast
 
 import toolz

ibis/backends/bigquery/udf/rewrite.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import ast
 from typing import Callable
 

ibis/backends/bigquery/version.py

@@ -1 +1,3 @@
+from __future__ import annotations
+
 __version__ = "2.2.0"

ibis/backends/clickhouse/compiler/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from public import public
 
 from ibis.backends.clickhouse.compiler.core import translate

ibis/backends/clickhouse/compiler/core.py

@@ -39,6 +39,8 @@
 those as input to construct the output.
 """
 
+from __future__ import annotations
+
 from typing import Any, Mapping
 
 import sqlglot as sg

ibis/backends/clickhouse/compiler/relations.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import functools
 from functools import partial
 

ibis/backends/clickhouse/identifiers.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 _identifiers = frozenset(
     {
         'add',

ibis/backends/clickhouse/tests/conftest.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import os
 from pathlib import Path
 from typing import Callable

ibis/backends/dask/client.py

@@ -1,5 +1,7 @@
 """The dask client implementation."""
 
+from __future__ import annotations
+
 import dask.dataframe as dd
 import numpy as np
 from pandas.api.types import DatetimeTZDtype

ibis/backends/dask/dispatch.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from multipledispatch import Dispatcher
 
 import ibis.backends.pandas.core as core_dispatch

ibis/backends/dask/execution/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from ibis.backends.dask.execution.aggregations import *  # noqa: F403
 from ibis.backends.dask.execution.arrays import *  # noqa: F403
 from ibis.backends.dask.execution.decimal import *  # noqa: F403

ibis/backends/dask/execution/arrays.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import itertools
 
 import dask.dataframe as dd

ibis/backends/dask/execution/decimal.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import decimal
 
 import dask.dataframe as dd

ibis/backends/dask/execution/indexing.py

@@ -1,5 +1,7 @@
 """Execution rules for ops.Where operations."""
 
+from __future__ import annotations
+
 import dask.dataframe as dd
 
 import ibis.expr.operations as ops

ibis/backends/dask/execution/join.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import dask.dataframe as dd
 from pandas import Timedelta
 

ibis/backends/dask/execution/maps.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from collections.abc import Mapping
 
 import dask.dataframe as dd

ibis/backends/dask/execution/numeric.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import collections
 import decimal
 import functools

ibis/backends/dask/execution/reductions.py

@@ -10,6 +10,8 @@
 dispatcher since the top level container is a list.
 """
 
+from __future__ import annotations
+
 import functools
 from collections.abc import Sized
 

ibis/backends/dask/execution/strings.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import functools
 import itertools
 import operator

ibis/backends/dask/execution/structs.py

@@ -1,5 +1,7 @@
 """Dask backend execution of struct fields and literals."""
 
+from __future__ import annotations
+
 import operator
 
 import dask.dataframe as dd

ibis/backends/dask/execution/temporal.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import datetime
 
 import dask.array as da

ibis/backends/dask/tests/conftest.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from pathlib import Path
 from typing import Any
 

ibis/backends/dask/tests/execution/conftest.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import decimal
 
 import pandas as pd

ibis/backends/dask/trace.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import functools
 import logging
 import traceback