diff --git a/src/sentry/tsdb/base.py b/src/sentry/tsdb/base.py index 44cf3f63c0316a..16845f10ca9cc2 100644 --- a/src/sentry/tsdb/base.py +++ b/src/sentry/tsdb/base.py @@ -419,7 +419,7 @@ def get_range( start: datetime, end: datetime, rollup: int | None = None, - environment_ids: list[int] | None = None, + environment_ids: Sequence[int] | None = None, conditions=None, use_cache: bool = False, jitter_value: int | None = None, @@ -477,13 +477,13 @@ def _add_jitter_to_series( return series def rollup( - self, values: Mapping[int, Sequence[tuple[float, int]]], rollup: int - ) -> dict[int, list[list[float]]]: + self, values: Mapping[TSDBKey, Sequence[tuple[float, int]]], rollup: int + ) -> dict[TSDBKey, list[list[float]]]: """ Given a set of values (as returned from ``get_range``), roll them up using the ``rollup`` time (in seconds). """ - result: dict[int, list[list[float]]] = {} + result: dict[TSDBKey, list[list[float]]] = {} for key, points in values.items(): result[key] = [] last_new_ts = None diff --git a/src/sentry/tsdb/dummy.py b/src/sentry/tsdb/dummy.py index ed9b5d7209bb48..7e57f8f4ba66a3 100644 --- a/src/sentry/tsdb/dummy.py +++ b/src/sentry/tsdb/dummy.py @@ -31,7 +31,7 @@ def get_range( start: datetime, end: datetime, rollup: int | None = None, - environment_ids: list[int] | None = None, + environment_ids: Sequence[int] | None = None, conditions=None, use_cache: bool = False, jitter_value: int | None = None, diff --git a/src/sentry/tsdb/inmemory.py b/src/sentry/tsdb/inmemory.py index dd2854c16000e6..1c78f6df4a7961 100644 --- a/src/sentry/tsdb/inmemory.py +++ b/src/sentry/tsdb/inmemory.py @@ -69,7 +69,7 @@ def get_range( start: datetime, end: datetime, rollup: int | None = None, - environment_ids: list[int] | None = None, + environment_ids: Sequence[int] | None = None, conditions=None, use_cache: bool = False, jitter_value: int | None = None, diff --git a/src/sentry/tsdb/redis.py b/src/sentry/tsdb/redis.py index 7c9a692c960482..d8972adb111c1b 100644 --- a/src/sentry/tsdb/redis.py +++ b/src/sentry/tsdb/redis.py @@ -316,7 +316,7 @@ def get_range( start: datetime, end: datetime, rollup: int | None = None, - environment_ids: list[int] | None = None, + environment_ids: Sequence[int] | None = None, conditions=None, use_cache: bool = False, jitter_value: int | None = None, diff --git a/src/sentry/tsdb/snuba.py b/src/sentry/tsdb/snuba.py index 5fe3bb838364fa..29fd5225efc9d8 100644 --- a/src/sentry/tsdb/snuba.py +++ b/src/sentry/tsdb/snuba.py @@ -710,7 +710,7 @@ def get_range( start: datetime, end: datetime, rollup: int | None = None, - environment_ids: list[int] | None = None, + environment_ids: Sequence[int] | None = None, conditions=None, use_cache: bool = False, jitter_value: int | None = None,