Fix typing more

PostHog · Apr 17, 2024 · bb87101 · bb87101
1 parent 65147a0
commit bb87101
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/posthog/api/query.py b/posthog/api/query.py
@@ -70,7 +70,7 @@ def create(self, request, *args, **kwargs) -> Response:
                 user_id=self.request.user.pk,
                 query_json=request.data["query"],
                 query_id=client_query_id,
-                refresh_requested=data.refresh,
+                refresh_requested=data.refresh or False,
             )
             return Response(query_status.model_dump(), status=status.HTTP_202_ACCEPTED)
 

diff --git a/posthog/hogql_queries/test/test_events_query_runner.py b/posthog/hogql_queries/test/test_events_query_runner.py
@@ -85,8 +85,9 @@ def _run_boolean_field_query(self, filter: EventPropertyFilter):
             )
 
             runner = EventsQueryRunner(query=query, team=self.team)
-            results = runner.run().results
-            assert isinstance(results, QueryResponse)
+            response = runner.run()
+            assert isinstance(response, QueryResponse)
+            results = response.results
             return results
 
     def test_is_not_set_boolean(self):