Remove unused methods

src/brad/routing/abstract_policy.py

@@ -14,14 +14,6 @@ class AbstractRoutingPolicy:
     def name(self) -> str:
         raise NotImplementedError
 
-    async def run_setup(self, estimator: Optional[Estimator] = None) -> None:
-        """
-        Should be called before using this policy. This is used to set up any
-        dynamic state.
-
-        If this routing policy needs an estimator, one should be provided here.
-        """
-
     async def engine_for(
         self, query_rep: QueryRep, ctx: RoutingContext
     ) -> List[Engine]:
@@ -65,17 +57,6 @@ def __init__(
         self.indefinite_policies = indefinite_policies
         self.definite_policy = definite_policy
 
-    async def run_setup(self, estimator: Optional[Estimator] = None) -> None:
-        """
-        Should be called before using the policy. This is used to set up any
-        dynamic state.
-
-        If this routing policy needs an estimator, one should be provided here.
-        """
-        for policy in self.indefinite_policies:
-            await policy.run_setup(estimator)
-        await self.definite_policy.run_setup(estimator)
-
     def __eq__(self, other: object):
         if not isinstance(other, FullRoutingPolicy):
             return False

src/brad/routing/tree_based/forest_policy.py

@@ -1,9 +1,8 @@
 import asyncio
-from typing import Optional, List, Dict, Any
+from typing import List, Dict, Any
 
 from brad.asset_manager import AssetManager
 from brad.config.engine import Engine
-from brad.data_stats.estimator import Estimator
 from brad.query_rep import QueryRep
 from brad.routing.abstract_policy import AbstractRoutingPolicy
 from brad.routing.context import RoutingContext
@@ -47,9 +46,6 @@ def __eq__(self, other: object) -> bool:
             return False
         return self._policy == other._policy and self._model == other._model
 
-    async def run_setup(self, estimator: Optional[Estimator] = None) -> None:
-        pass
-
     async def engine_for(
         self, query_rep: QueryRep, ctx: RoutingContext
     ) -> List[Engine]: