From 2add6165ce2deff7d841b6cf990323ab7c4b6742 Mon Sep 17 00:00:00 2001 From: David Tsukernik Date: Thu, 7 Nov 2024 15:58:52 -0800 Subject: [PATCH] Fix test error --- snuba/web/rpc/common/common.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/snuba/web/rpc/common/common.py b/snuba/web/rpc/common/common.py index c6af39f058..56b03fe458 100644 --- a/snuba/web/rpc/common/common.py +++ b/snuba/web/rpc/common/common.py @@ -127,12 +127,8 @@ def aggregation_to_expression(aggregation: AttributeAggregation) -> Expression: == ExtrapolationMode.EXTRAPOLATION_MODE_SAMPLE_WEIGHTED ): agg_func_expr = function_map_sample_weighted.get(aggregation.aggregate) - elif aggregation.extrapolation_mode == ExtrapolationMode.EXTRAPOLATION_MODE_NONE: - agg_func_expr = function_map.get(aggregation.aggregate) else: - raise BadSnubaRPCRequestException( - f"Unsupported extrapolation mode: {aggregation.extrapolation_mode}" - ) + agg_func_expr = function_map.get(aggregation.aggregate) if agg_func_expr is None: raise BadSnubaRPCRequestException(