chore: allow pyupgrade to upgrade type annotations that were required…

… by python 3.8

ci/check_disallowed_imports.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python3
+from __future__ import annotations
 
 import collections
 import fnmatch

docs/backends/app/backend_info_app.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import datetime
 import tempfile
 from pathlib import Path

docs/blog/rendered/model.py

@@ -2,6 +2,7 @@
 
 Adapted from https://gist.github.com/pdet/e8d38734232c08e6c15aba79b4eb8368#file-taxi_prediction_example-py.
 """
+from __future__ import annotations
 
 import pyarrow as pa
 import torch

docs/example_streamlit_app/example_streamlit_app.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import requests
 import streamlit as st
 

gen_redirects.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pathlib
 
 import mkdocs_gen_files

ibis/backends/bigquery/custom_udfs.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import ibis.expr.datatypes as dt
 import ibis.expr.operations as ops
 from ibis.backends.bigquery.compiler import BigQueryExprTranslator

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

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

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

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from unittest import mock
 
 import google.api_core.client_options

ibis/backends/bigquery/tests/system/udf/test_udf_execute.py

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

ibis/backends/bigquery/tests/unit/test_client.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pytest
 
 from ibis.backends.bigquery import client

ibis/backends/bigquery/tests/unit/test_compiler.py

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

ibis/backends/bigquery/tests/unit/test_datatypes.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pytest
 from pytest import param
 

ibis/backends/bigquery/tests/unit/udf/test_core.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import builtins
 import sys
 import tempfile

ibis/backends/bigquery/tests/unit/udf/test_find.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import ast
 
 from ibis.backends.bigquery.udf.find import find_names

ibis/backends/bigquery/tests/unit/udf/test_usage.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pytest
 from pytest import param
 

ibis/backends/clickhouse/tests/test_aggregations.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from operator import methodcaller
 
 import numpy as np

ibis/backends/clickhouse/tests/test_client.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pandas as pd
 import pandas.testing as tm
 import pyarrow as pa

ibis/backends/clickhouse/tests/test_functions.py

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

ibis/backends/clickhouse/tests/test_literals.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pytest
 from pandas import Timestamp
 from pytest import param

ibis/backends/clickhouse/tests/test_operators.py

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

ibis/backends/clickhouse/tests/test_select.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pandas as pd
 import pandas.testing as tm
 import pytest

ibis/backends/clickhouse/tests/test_types.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pytest
 from pytest import param
 

ibis/backends/dask/aggcontext.py

@@ -1,8 +1,9 @@
+from __future__ import annotations
+
 import operator
-from typing import Any, Callable, Dict, Tuple, Union
+from typing import TYPE_CHECKING, Any, Callable, Dict, Tuple, Union
 
 import dask.dataframe as dd
-from dask.dataframe.groupby import SeriesGroupBy
 
 import ibis
 from ibis.backends.pandas.aggcontext import (
@@ -15,6 +16,9 @@
 )
 from ibis.backends.pandas.aggcontext import Transform as PandasTransform
 
+if TYPE_CHECKING:
+    from dask.dataframe.groupby import SeriesGroupBy
+
 # TODO Consolidate this logic with the pandas aggcontext.
 # This file is almost a direct port of the pandas aggcontext.
 # https://github.com/ibis-project/ibis/issues/5911

ibis/backends/dask/core.py

@@ -437,6 +437,7 @@ def execute_and_reset(
     kwargs : Dict[str, object]
         Additional arguments that can potentially be used by individual node
         execution
+
     Returns
     -------
     result : Union[

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

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import operator
 
 import numpy as np

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

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

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

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

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

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pandas as pd
 import pytest
 from pandas import Timedelta, date_range

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

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import numpy as np
 import pandas as pd
 import pytest

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

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

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

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from warnings import catch_warnings
 
 import pytest

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

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

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

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

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

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pytest
 
 from ibis.backends.dask.execution.util import assert_identical_grouping_keys

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

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import io
 from datetime import date
 from operator import methodcaller

ibis/backends/dask/tests/test_client.py

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

ibis/backends/dask/tests/test_core.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pytest
 from dask.dataframe.utils import tm
 

ibis/backends/dask/tests/test_dispatcher.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pytest
 from multipledispatch import Dispatcher
 

ibis/backends/dask/tests/test_udf.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import collections
 
 import numpy as np

ibis/backends/datafusion/tests/test_register.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pandas as pd
 import pyarrow as pa
 import pyarrow.dataset as ds

ibis/backends/datafusion/tests/test_select.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pytest
 
 from ibis.backends.datafusion.tests.conftest import BackendTest

ibis/backends/datafusion/tests/test_udf.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pandas.testing as tm
 import pytest
 

ibis/backends/duckdb/tests/test_datatypes.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import duckdb_engine
 import pytest
 import sqlalchemy as sa

ibis/backends/duckdb/tests/test_register.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import os
 import sqlite3
 import tempfile

ibis/backends/flink/registry.py

@@ -1,11 +1,17 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
 import ibis.expr.operations as ops
-from ibis.backends.base.sql.compiler import ExprTranslator
 from ibis.backends.base.sql.registry import helpers
 from ibis.backends.base.sql.registry import (
     operation_registry as base_operation_registry,
 )
 from ibis.common.temporal import TimestampUnit
 
+if TYPE_CHECKING:
+    from ibis.backends.base.sql.compiler import ExprTranslator
+
 operation_registry = base_operation_registry.copy()
 
 

ibis/backends/flink/tests/conftest.py

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

ibis/backends/flink/tests/test_join.py

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

ibis/backends/flink/tests/test_literals.py

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

ibis/backends/flink/tests/test_translator.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pytest
 from pytest import param
 

ibis/backends/flink/translator.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from ibis.backends.base.sql.compiler import ExprTranslator
 from ibis.backends.flink.registry import operation_registry
 

ibis/backends/flink/utils.py

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

ibis/backends/impala/tests/test_analytic_functions.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pytest
 
 from ibis.backends.impala.tests.conftest import translate

ibis/backends/impala/tests/test_bucket_histogram.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pytest
 
 from ibis.backends.impala.compiler import ImpalaCompiler

ibis/backends/impala/tests/test_case_exprs.py

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

ibis/backends/impala/tests/test_client.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import datetime
 from contextlib import closing
 from posixpath import join as pjoin

ibis/backends/impala/tests/test_coalesce_greater_least.py

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

ibis/backends/impala/tests/test_ddl.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import os
 from contextlib import closing
 from posixpath import join as pjoin

ibis/backends/impala/tests/test_ddl_compilation.py

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

ibis/backends/impala/tests/test_exprs.py

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

ibis/backends/impala/tests/test_in_not_in.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pytest
 
 from ibis import literal as L

ibis/backends/impala/tests/test_metadata.py

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