From 579a9caa4bfab58c3caf23ac75ceb5762b1ff676 Mon Sep 17 00:00:00 2001 From: Ted Kaemming <65315+tkaemming@users.noreply.github.com> Date: Thu, 14 Dec 2023 18:47:07 -0800 Subject: [PATCH] Fix typing in tests --- .../temporal/tests/test_squash_person_overrides_workflow.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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