style(lint): Auto commit lint changes

snuba/web/db_query.py

@@ -2,7 +2,6 @@
 
 import logging
 import random
-from snuba.datasets.storages.storage_key import StorageKey
 import uuid
 from concurrent.futures import ThreadPoolExecutor
 from dataclasses import dataclass
@@ -26,6 +25,7 @@
 from snuba.clickhouse.query import Query
 from snuba.clickhouse.query_dsl.accessors import get_time_range_estimate
 from snuba.clickhouse.query_profiler import generate_profile
+from snuba.datasets.storages.storage_key import StorageKey
 from snuba.query import ProcessableQuery
 from snuba.query.allocation_policies import (
     MAX_THRESHOLD,
@@ -682,15 +682,24 @@ def visit_join_clause(self, node: JoinClause[Table]) -> None:
         node.right_node.accept(self)
 
 
-def _record_bytes_scanned(result_or_error: QueryResultOrError, attribution_info: AttributionInfo, dataset_name: str, storage_key: StorageKey):
+def _record_bytes_scanned(
+    result_or_error: QueryResultOrError,
+    attribution_info: AttributionInfo,
+    dataset_name: str,
+    storage_key: StorageKey,
+):
     custom_metrics = MetricsWrapper(environment.metrics, "allocation_policy")
 
     if result_or_error.query_result:
         progress_bytes_scanned = cast(int, result_or_error.query_result.result.get("profile", {}).get("progress_bytes", 0))  # type: ignore
         custom_metrics.increment(
             "bytes_scanned",
             progress_bytes_scanned,
-            tags={"referrer": attribution_info.referrer, "dataset": dataset_name, "storage_key": storage_key.value},
+            tags={
+                "referrer": attribution_info.referrer,
+                "dataset": dataset_name,
+                "storage_key": storage_key.value,
+            },
         )
 
 
@@ -806,12 +815,17 @@ def db_query(
         raise e
     finally:
         result_or_error = QueryResultOrError(query_result=result, error=error)
-        _record_bytes_scanned(result_or_error, attribution_info, dataset_name, allocation_policies[0].storage_key)
+        _record_bytes_scanned(
+            result_or_error,
+            attribution_info,
+            dataset_name,
+            allocation_policies[0].storage_key,
+        )
         for allocation_policy in allocation_policies:
             allocation_policy.update_quota_balance(
                 tenant_ids=attribution_info.tenant_ids,
                 query_id=query_id,
-                result_or_error=result_or_error
+                result_or_error=result_or_error,
             )
         if stats.get("cache_hit"):
             metrics.increment("cache_hit", tags={"dataset": dataset_name})