Fix minor issues

PostHog · Mar 19, 2024 · 8e12666 · 8e12666
1 parent 6a6104c
commit 8e12666
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/posthog/hogql_queries/insights/funnels/base.py b/posthog/hogql_queries/insights/funnels/base.py
@@ -202,7 +202,7 @@ def breakdown_values(self) -> List[int] | List[str] | List[List[str]]:
                 )
 
             # execute query
-            results = execute_hogql_query(values_query, self.context.team, limit_context=self.limit_context).results
+            results = execute_hogql_query(values_query, self.context.team).results
             if results is None:
                 raise ValidationError("Apologies, there has been an error computing breakdown values.")
             return [row[0] for row in results[0:breakdown_limit_or_default]]

diff --git a/posthog/hogql_queries/insights/test/test_paginators.py b/posthog/hogql_queries/insights/test/test_paginators.py
@@ -1,3 +1,5 @@
+from typing import cast
+from posthog.hogql.ast import SelectQuery
 from posthog.hogql.constants import (
     LimitContext,
     get_default_limit_for_context,
@@ -136,7 +138,7 @@ def test_response_params_consistency(self):
         """Test consistency of response_params method."""
         paginator = HogQLHasMorePaginator(limit=5, offset=10)
         paginator.response = paginator.execute_hogql_query(
-            parse_select("SELECT * FROM persons"),
+            cast(SelectQuery, parse_select("SELECT * FROM persons")),
             query_type="test_query",
             team=self.team,
         )