diff --git a/target_postgres/tests/test_target_postgres.py b/target_postgres/tests/test_target_postgres.py index fd6656aa..843e29ab 100644 --- a/target_postgres/tests/test_target_postgres.py +++ b/target_postgres/tests/test_target_postgres.py @@ -2,6 +2,7 @@ # flake8: noqa import copy import io +import typing as t from contextlib import redirect_stdout from decimal import Decimal from pathlib import Path @@ -93,8 +94,8 @@ def verify_data( self, table_name: str, number_of_rows: int = 1, - primary_key: str | None = None, - check_data: dict | list[dict] | None = None, + primary_key: t.Union[str, None] = None, + check_data: t.Union[dict, list[dict], None] = None, ): """Checks whether the data in a table matches a provided data sample.