diff --git a/posthog/temporal/tests/test_squash_person_overrides_workflow.py b/posthog/temporal/tests/test_squash_person_overrides_workflow.py index 3b8ee34c404480..899647f84b69b9 100644 --- a/posthog/temporal/tests/test_squash_person_overrides_workflow.py +++ b/posthog/temporal/tests/test_squash_person_overrides_workflow.py @@ -2,7 +2,7 @@ import random from collections import defaultdict from datetime import datetime, timedelta -from typing import NamedTuple, TypedDict +from typing import Iterator, NamedTuple, TypedDict from uuid import UUID, uuid4 import psycopg2 @@ -943,7 +943,7 @@ class PostgresPersonOverrideFixtures(NamedTuple): @pytest.fixture(params=POSTGRES_PERSON_OVERRIDES_MANAGERS.keys()) def postgres_person_override_fixtures( request, query_inputs: QueryInputs, team_id, pg_connection -) -> PersonOverrideTuple: +) -> Iterator[PersonOverrideTuple]: """Create a PersonOverrideMapping and a PersonOverride. We cannot use the Django ORM safely in an async context, so we INSERT INTO directly