diff --git a/src/brad/front_end/front_end.py b/src/brad/front_end/front_end.py index 5771b300..ad4259d3 100644 --- a/src/brad/front_end/front_end.py +++ b/src/brad/front_end/front_end.py @@ -90,13 +90,18 @@ def __init__( from brad.front_end.flight_sql_server import BradFlightSqlServer self._flight_sql_server: Optional[BradFlightSqlServer] = ( - BradFlightSqlServer(host="0.0.0.0", - port=31337, - callback=self._handle_query_from_flight_sql) + BradFlightSqlServer( + host="0.0.0.0", + port=31337, + callback=self._handle_query_from_flight_sql, + ) ) + self._flight_sql_server_session_id = None else: self._flight_sql_server = None + self._main_thread_loop = None + self._fe_index = fe_index self._config = config self._schema_name = schema_name @@ -195,7 +200,7 @@ def __init__( def _handle_query_from_flight_sql(self, query: str) -> RowList: future = asyncio.run_coroutine_threadsafe( self._run_query_impl(self._flight_sql_server_session_id, query, {}), - self._main_thread_loop + self._main_thread_loop, ) row_result = future.result()