Fix bug in stub mode detection

mitdbg · Mar 30, 2024 · 0a03728 · 0a03728
1 parent cfda1e7
commit 0a03728
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/brad/front_end/front_end.py b/src/brad/front_end/front_end.py
@@ -188,7 +188,7 @@ def __init__(
         )
         self._ping_watchdog_task: Optional[asyncio.Task[None]] = None
 
-        self._is_stub_mode = self._config.stub_mode_path is not None
+        self._is_stub_mode = self._config.stub_mode_path() is not None
 
     async def serve_forever(self):
         await self._run_setup()

diff --git a/src/brad/front_end/session.py b/src/brad/front_end/session.py
@@ -116,7 +116,7 @@ async def create_new_session(self) -> Tuple[SessionId, Session]:
             routing_policy_override == RoutingPolicy.ForestTableSelectivity
             or routing_policy_override == RoutingPolicy.Default
         ):
-            if self._config.stub_mode_path is None:
+            if self._config.stub_mode_path() is not None:
                 estimator = await PostgresEstimator.connect(
                     self._schema_name, self._config
                 )