ref: improve signatures of tsdb (#74725)

<!-- Describe your PR here. -->

Author: asottile-sentry

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

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,

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,

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,

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,