Address python type check errors

mitdbg · Apr 16, 2024 · 7d85b67 · 7d85b67
1 parent a76dc0d
commit 7d85b67
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/brad/front_end/front_end.py b/src/brad/front_end/front_end.py
@@ -96,11 +96,11 @@ def __init__(
                     callback=self._handle_query_from_flight_sql,
                 )
             )
-            self._flight_sql_server_session_id = None
+            self._flight_sql_server_session_id: Optional[SessionId] = None
         else:
             self._flight_sql_server = None
 
-        self._main_thread_loop = None
+        self._main_thread_loop: Optional[AbstractEventLoop] = None
 
         self._fe_index = fe_index
         self._config = config
@@ -198,6 +198,8 @@ def __init__(
         self._is_stub_mode = self._config.stub_mode_path is not None
 
     def _handle_query_from_flight_sql(self, query: str) -> RowList:
+        assert self._flight_sql_server_session_id is not None
+
         future = asyncio.run_coroutine_threadsafe(
             self._run_query_impl(self._flight_sql_server_session_id, query, {}),
             self._main_thread_loop,