Add from __future__ import annotations

That is for files with pre-3.9 annotations with view to migrate
Incidentally run isort (see unionai-oss#1295)

Signed-off-by: Nathan McDougall <[email protected]>

noxfile.py

@@ -1,22 +1,21 @@
 """Nox sessions."""
+from __future__ import annotations
+
 # isort: skip_file
 import os
 import re
 import shutil
 import sys
+from distutils.core import run_setup
 from typing import Dict, List
 
+import nox
+
 # setuptools must be imported before distutils !
 import setuptools
-from distutils.core import (
-    run_setup,
-)
-
-import nox
 from nox import Session
 from pkg_resources import Requirement, parse_requirements
 
-
 nox.options.sessions = (
     "requirements",
     "mypy",

pandera/accessors/pandas_accessor.py

@@ -1,4 +1,5 @@
 """Register pandas accessor for pandera schema metadata."""
+from __future__ import annotations
 
 from typing import Optional, Union
 

pandera/accessors/pyspark_sql_accessor.py

@@ -1,5 +1,6 @@
 """Custom accessor functionality for PySpark.Sql. Register pyspark accessor for pandera schema metadata.
 """
+from __future__ import annotations
 
 import warnings
 from typing import Optional

pandera/api/base/checks.py

@@ -1,4 +1,5 @@
 """Data validation base check."""
+from __future__ import annotations
 
 import inspect
 from itertools import chain

pandera/api/base/model.py

@@ -1,4 +1,5 @@
 """Base classes for model api."""
+from __future__ import annotations
 
 import os
 from typing import (

pandera/api/base/model_components.py

@@ -1,4 +1,5 @@
 """Model component base classes."""
+from __future__ import annotations
 
 from typing import (
     Any,

pandera/api/base/model_config.py

@@ -1,4 +1,5 @@
 """Class-based dataframe model API configuration."""
+from __future__ import annotations
 
 from typing import Any, Optional
 

pandera/api/base/schema.py

@@ -4,15 +4,16 @@
 data validation. These operations are exposed as methods that are composed
 together to implement the pandera schema specification.
 """
+from __future__ import annotations
 
 import inspect
 from abc import ABC
 from functools import wraps
-from typing import Any, Dict, Tuple, Type, Optional, Union
+from typing import Any, Dict, Optional, Tuple, Type, Union
 
 from pandera.backends.base import BaseSchemaBackend
-from pandera.errors import BackendNotFoundError
 from pandera.dtypes import DataType
+from pandera.errors import BackendNotFoundError
 
 DtypeInputTypes = Union[str, type, DataType, Type]
 

pandera/api/checks.py

@@ -1,4 +1,5 @@
 """Data validation check definition."""
+from __future__ import annotations
 
 import re
 from typing import (

pandera/api/extensions.py

@@ -1,4 +1,5 @@
 """Extensions module."""
+from __future__ import annotations
 
 import inspect
 import warnings

pandera/api/hypotheses.py

@@ -1,4 +1,5 @@
 """Data validation checks for hypothesis testing."""
+from __future__ import annotations
 
 from typing import Any, Callable, Dict, List, Optional, TypeVar, Union
 

pandera/api/pandas/array.py

@@ -1,4 +1,5 @@
 """Core pandas array specification."""
+from __future__ import annotations
 
 import copy
 import warnings

pandera/api/pandas/components.py

@@ -1,4 +1,5 @@
 """Core pandas schema component specifications."""
+from __future__ import annotations
 
 import warnings
 from typing import Any, Dict, Iterable, List, Optional, Tuple, Union, cast

pandera/api/pandas/model.py

@@ -1,4 +1,5 @@
 """Class-based api for pandas models."""
+from __future__ import annotations
 
 import copy
 import inspect

pandera/api/pandas/model_components.py

@@ -1,4 +1,6 @@
 """DataFrameModel components"""
+from __future__ import annotations
+
 from typing import (
     Any,
     Callable,

pandera/api/pandas/model_config.py

@@ -1,4 +1,5 @@
 """Class-based dataframe model API configuration for pandas."""
+from __future__ import annotations
 
 from typing import Any, Callable, Dict, List, Optional, Union
 

pandera/api/pyspark/column_schema.py

@@ -1,4 +1,5 @@
 """Core pyspark column specification."""
+from __future__ import annotations
 
 import copy
 from typing import Any, List, Optional, TypeVar, cast

pandera/api/pyspark/components.py

@@ -1,4 +1,5 @@
 """Core pyspark schema component specifications."""
+from __future__ import annotations
 
 from typing import Any, Dict, Iterable, Optional, Tuple, Union
 

pandera/api/pyspark/error_handler.py

@@ -1,4 +1,5 @@
 """Handle schema errors."""
+from __future__ import annotations
 
 from collections import defaultdict
 from enum import Enum

pandera/api/pyspark/model_components.py

@@ -1,4 +1,6 @@
 """DataFrameModel components"""
+from __future__ import annotations
+
 from typing import (
     Any,
     Callable,

pandera/backends/pandas/builtin_hypotheses.py

@@ -1,11 +1,13 @@
-# pylint: disable=missing-function-docstring
 """Pandas implementation of built-in hypotheses."""
+from __future__ import annotations
 
 from typing import Tuple
 
 from pandera.api.extensions import register_builtin_hypothesis
 from pandera.backends.pandas.builtin_checks import PandasData
 
+# pylint: disable=missing-function-docstring
+
 
 @register_builtin_hypothesis(
     error="failed two sample ttest between '{sample1}' and '{sample2}'",

pandera/backends/pandas/components.py

@@ -1,5 +1,5 @@
 """Backend implementation for pandas schema components."""
-# pylint: disable=too-many-locals
+from __future__ import annotations
 
 import traceback
 from copy import copy, deepcopy
@@ -8,24 +8,26 @@
 import numpy as np
 import pandas as pd
 
-from pandera.backends.base import CoreCheckResult
-from pandera.backends.pandas.array import ArraySchemaBackend
-from pandera.backends.pandas.container import DataFrameSchemaBackend
 from pandera.api.pandas.types import (
     is_field,
     is_index,
     is_multiindex,
     is_table,
 )
+from pandera.backends.base import CoreCheckResult
+from pandera.backends.pandas.array import ArraySchemaBackend
+from pandera.backends.pandas.container import DataFrameSchemaBackend
 from pandera.backends.pandas.error_formatters import scalar_failure_case
 from pandera.error_handlers import SchemaErrorHandler
 from pandera.errors import (
+    SchemaDefinitionError,
     SchemaError,
-    SchemaErrors,
     SchemaErrorReason,
-    SchemaDefinitionError,
+    SchemaErrors,
 )
 
+# pylint: disable=too-many-locals
+
 
 class ColumnBackend(ArraySchemaBackend):
     """Backend implementation for pandas dataframe columns."""

pandera/backends/pandas/container.py

@@ -1,22 +1,23 @@
 """Pandas Parsing, Validation, and Error Reporting Backends."""
+from __future__ import annotations
 
 import copy
 import itertools
 import traceback
-from typing import Any, Callable, List, Optional, Tuple, Dict
+from typing import Any, Callable, Dict, List, Optional, Tuple
 
 import pandas as pd
 from pydantic import BaseModel
 
-from pandera.backends.base import CoreCheckResult
 from pandera.api.pandas.types import is_table
+from pandera.backends.base import CoreCheckResult
 from pandera.backends.pandas.base import ColumnInfo, PandasSchemaBackend
 from pandera.backends.pandas.error_formatters import (
     reshape_failure_cases,
     scalar_failure_case,
 )
-from pandera.engines import pandas_engine
 from pandera.backends.pandas.utils import convert_uniquesettings
+from pandera.engines import pandas_engine
 from pandera.error_handlers import SchemaErrorHandler
 from pandera.errors import (
     ParserError,

pandera/backends/pandas/hypotheses.py

@@ -1,4 +1,5 @@
 """Hypothesis backend for pandas."""
+from __future__ import annotations
 
 from functools import partial
 from typing import Any, Callable, Dict, Union, cast

pandera/backends/pyspark/base.py

@@ -1,4 +1,5 @@
 """pyspark Parsing, Validation, and Error Reporting Backends."""
+from __future__ import annotations
 
 import warnings
 from typing import (

pandera/backends/pyspark/checks.py

@@ -1,4 +1,5 @@
 """Check backend for pyspark."""
+from __future__ import annotations
 
 from functools import partial
 from typing import Dict, List, Optional

pandera/backends/pyspark/container.py

@@ -1,17 +1,18 @@
 """Pyspark Parsing, Validation, and Error Reporting Backends."""
+from __future__ import annotations
 
 import copy
 import traceback
 import warnings
-from typing import Any, List, Optional, Dict
+from typing import Any, Dict, List, Optional
 
 from pyspark.sql import DataFrame
 from pyspark.sql.functions import col
 
 from pandera.api.pyspark.error_handler import ErrorCategory, ErrorHandler
 from pandera.api.pyspark.types import is_table
 from pandera.backends.pyspark.base import ColumnInfo, PysparkSchemaBackend
-from pandera.backends.pyspark.decorators import validate_scope, ValidationScope
+from pandera.backends.pyspark.decorators import ValidationScope, validate_scope
 from pandera.backends.pyspark.error_formatters import scalar_failure_case
 from pandera.config import CONFIG
 from pandera.errors import (

pandera/decorators.py

@@ -1,4 +1,6 @@
 """Decorators for integrating pandera into existing data pipelines."""
+from __future__ import annotations
+
 import functools
 import inspect
 import sys

pandera/engines/engine.py

@@ -1,4 +1,6 @@
 """Data types engine interface."""
+from __future__ import annotations
+
 # https://github.com/PyCQA/pylint/issues/3268
 # pylint:disable=no-value-for-parameter
 import functools
@@ -25,7 +27,6 @@
 
 from pandera.dtypes import DataType
 
-
 # register different TypedDict type depending on python version
 if sys.version_info >= (3, 9):
     from typing import TypedDict

pandera/engines/pandas_engine.py

@@ -1,11 +1,6 @@
 """Pandas engine and data types."""
-# pylint:disable=too-many-ancestors
-
-# docstrings are inherited
-# pylint:disable=missing-class-docstring
+from __future__ import annotations
 
-# pylint doesn't know about __init__ generated with dataclass
-# pylint:disable=unexpected-keyword-arg,no-value-for-parameter
 import builtins
 import dataclasses
 import datetime
@@ -42,6 +37,14 @@
 from pandera.engines.utils import pandas_version
 from pandera.system import FLOAT_128_AVAILABLE
 
+# pylint doesn't know about __init__ generated with dataclass
+# pylint:disable=unexpected-keyword-arg,no-value-for-parameter
+# pylint:disable=too-many-ancestors
+
+# docstrings are inherited
+# pylint:disable=missing-class-docstring
+
+
 try:
     import pyarrow  # pylint: disable=unused-import
 

pandera/engines/pyspark_engine.py

@@ -1,18 +1,12 @@
 """PySpark engine and data types."""
-# pylint:disable=too-many-ancestors,no-member
-
-# docstrings are inherited
-# pylint:disable=missing-class-docstring
-
-# pylint doesn't know about __init__ generated with dataclass
-# pylint:disable=unexpected-keyword-arg,no-value-for-parameter
+from __future__ import annotations
 
 import dataclasses
 import inspect
 import re
-import warnings
-from typing import Any, Iterable, Union, Optional
 import sys
+import warnings
+from typing import Any, Iterable, Optional, Union
 
 import pyspark.sql.types as pst
 
@@ -21,6 +15,15 @@
 from pandera.engines import engine
 from pandera.engines.type_aliases import PysparkObject
 
+# pylint:disable=too-many-ancestors,no-member
+
+# docstrings are inherited
+# pylint:disable=missing-class-docstring
+
+# pylint doesn't know about __init__ generated with dataclass
+# pylint:disable=unexpected-keyword-arg,no-value-for-parameter
+
+
 try:
     import pyarrow  # pylint:disable=unused-import
 

pandera/engines/utils.py

@@ -1,4 +1,5 @@
 """Engine module utilities."""
+from __future__ import annotations
 
 from typing import Any, Union
 

pandera/error_handlers.py

@@ -1,8 +1,9 @@
 """Handle schema errors."""
+from __future__ import annotations
 
 from typing import List, Optional
 
-from pandera.errors import SchemaError, SchemaErrors, SchemaErrorReason
+from pandera.errors import SchemaError, SchemaErrorReason, SchemaErrors
 
 
 class SchemaErrorHandler:

pandera/errors.py

@@ -1,4 +1,5 @@
 """pandera-specific errors."""
+from __future__ import annotations
 
 import warnings
 from enum import Enum

pandera/mypy.py

@@ -1,4 +1,5 @@
 """Pandera mypy plugin."""
+from __future__ import annotations
 
 from typing import Callable, Optional, Union, cast