Unverified Commit f435da14 authored by Hongkuan Zhou's avatar Hongkuan Zhou Committed by GitHub
Browse files

fix: set correct component type in agg planner (#7176)


Signed-off-by: default avatarhongkuanz <hongkuanz@nvidia.com>
parent b7224eb5
......@@ -64,9 +64,8 @@ class AggPlanner:
shared_state=self.shared_state,
prometheus_metrics=prometheus_metrics,
start_prometheus_server=True,
component_type=SubComponentType.DECODE,
)
# Override: agg planner uses component_type DECODE for metrics fetching
self.planner.component_type = SubComponentType.DECODE
# Create both regression models (agg needs both TTFT and ITL)
self.ttft_regression = LoadBasedRegressionModel(
......
......@@ -257,7 +257,11 @@ class BasePlanner:
prometheus_engine_client: Optional[DirectRouterMetricsClient] = None,
connector=None,
start_prometheus_server: bool = True,
component_type: Optional[SubComponentType] = None,
):
if component_type is not None:
self.component_type = component_type
self.config = config
self.dryrun = dryrun
self.shared_state = shared_state or PlannerSharedState()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment