mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-04 18:13:44 +00:00
fix(tests): resolve linting issues in telemetry tests
This commit is contained in:
parent
a8a8aa56c0
commit
583df48479
3 changed files with 27 additions and 19 deletions
|
|
@ -6,7 +6,7 @@
|
|||
|
||||
"""Shared helpers for telemetry test collectors."""
|
||||
|
||||
from collections.abc import Iterable
|
||||
from collections.abc import Iterable, Mapping
|
||||
from dataclasses import dataclass
|
||||
from typing import Any
|
||||
|
||||
|
|
@ -14,7 +14,7 @@ from typing import Any
|
|||
@dataclass
|
||||
class SpanStub:
|
||||
name: str
|
||||
attributes: dict[str, Any]
|
||||
attributes: Mapping[str, Any] | None = None
|
||||
resource_attributes: dict[str, Any] | None = None
|
||||
events: list[dict[str, Any]] | None = None
|
||||
trace_id: str | None = None
|
||||
|
|
|
|||
|
|
@ -55,9 +55,12 @@ class InMemoryTelemetryCollector(BaseTelemetryCollector):
|
|||
def _snapshot_metrics(self) -> Any | None:
|
||||
data = self._metric_reader.get_metrics_data()
|
||||
if data and data.resource_metrics:
|
||||
resource_metric = data.resource_metrics[0]
|
||||
if resource_metric.scope_metrics:
|
||||
return resource_metric.scope_metrics[0].metrics
|
||||
all_metrics = []
|
||||
for resource_metric in data.resource_metrics:
|
||||
if resource_metric.scope_metrics:
|
||||
for scope_metric in resource_metric.scope_metrics:
|
||||
all_metrics.extend(scope_metric.metrics)
|
||||
return all_metrics if all_metrics else None
|
||||
return None
|
||||
|
||||
def _clear_impl(self) -> None:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue