fix: Update PersonDistinctIdOverridesTable to be consistent with up…

…dated `LazyTable` interface
PostHog · Mar 22, 2024 · a7fe876 · a7fe876
1 parent 060ebc2
commit a7fe876
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/posthog/hogql/database/schema/person_distinct_id_overrides.py b/posthog/hogql/database/schema/person_distinct_id_overrides.py
@@ -14,7 +14,6 @@
 )
 from posthog.hogql.database.schema.persons import join_with_persons_table
 from posthog.hogql.errors import HogQLException
-from posthog.schema import HogQLQueryModifiers
 
 PERSON_DISTINCT_ID_OVERRIDES_FIELDS = {
     "team_id": IntegerDatabaseField(name="team_id"),
@@ -82,7 +81,7 @@ def to_printed_hogql(self):
 class PersonDistinctIdOverridesTable(LazyTable):
     fields: Dict[str, FieldOrTable] = PERSON_DISTINCT_ID_OVERRIDES_FIELDS
 
-    def lazy_select(self, requested_fields: Dict[str, List[str | int]], modifiers: HogQLQueryModifiers):
+    def lazy_select(self, requested_fields: Dict[str, List[str | int]], context: HogQLContext, node: SelectQuery):
         return select_from_person_distinct_id_overrides_table(requested_fields)
 
     def to_printed_clickhouse(self, context):