add import fix

chispa/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import os
 import sys
 from glob import glob

chispa/bcolors.py

@@ -1,3 +1,6 @@
+from __future__ import annotations
+
+
 class bcolors:
     NC = "\033[0m"  # No Color, reset all
 

chispa/column_comparer.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from prettytable import PrettyTable
 
 from chispa.bcolors import bcolors

chispa/dataframe_comparer.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from functools import reduce
 
 from chispa.formatting.formats import FormattingConfig

chispa/formatting/__init__.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from chispa.formatting.formats import Color, Format, FormattingConfig, Style
 from chispa.formatting.terminal_string_formatter import TerminalStringFormatter
 

chispa/formatting/formats.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from dataclasses import dataclass, field
 from enum import Enum
 

chispa/number_helpers.py

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

chispa/row_comparer.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import math
 
 from pyspark.sql import Row

chispa/rows_comparer.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from itertools import zip_longest
 
 from prettytable import PrettyTable

chispa/schema_comparer.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from itertools import zip_longest
 
 from prettytable import PrettyTable

chispa/structfield_comparer.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from chispa.schema_comparer import are_structfields_equal
 
 __all__ = ("are_structfields_equal",)

docs/gen_ref_pages.py

@@ -3,6 +3,8 @@
 https://mkdocstrings.github.io/recipes/#automatic-code-reference-pages
 """
 
+from __future__ import annotations
+
 from pathlib import Path
 
 import mkdocs_gen_files

tests/conftest.py

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

tests/formatting/test_formats.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from chispa.formatting.formats import Color, FormattingConfig, Style
 
 

tests/formatting/test_terminal_string_formatter.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pytest
 
 from chispa.formatting.formats import Color, Format, Style

tests/spark.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from pyspark.sql import SparkSession
 
 spark = SparkSession.builder.master("local").appName("chispa").getOrCreate()

tests/test_column_comparer.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pytest
 
 from chispa import ColumnsNotEqualError, assert_approx_column_equality, assert_column_equality

tests/test_dataframe_comparer.py

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

tests/test_readme_examples.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pyspark.sql.functions as F
 import pytest
 from pyspark.sql import SparkSession

tests/test_row_comparer.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from pyspark.sql import Row
 
 from chispa.row_comparer import are_rows_approx_equal, are_rows_equal, are_rows_equal_enhanced

tests/test_schema_comparer.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import pytest
 from pyspark.sql.types import ArrayType, DoubleType, IntegerType, StringType, StructField, StructType
 

tests/test_structfield_comparer.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from pyspark.sql.types import DoubleType, IntegerType, StructField, StructType
 
 from chispa.structfield_comparer import are_structfields_equal