@@ -51,7 +51,7 @@ def test_numeric_query(self):
5151 query , _ = self .resolver .resolve_query ("ai.total_tokens.used:123" )
5252 assert query == TraceItemFilter (
5353 comparison_filter = ComparisonFilter (
54- key = AttributeKey (name = "ai_total_tokens_used" , type = AttributeKey .Type .TYPE_FLOAT ),
54+ key = AttributeKey (name = "ai_total_tokens_used" , type = AttributeKey .Type .TYPE_DOUBLE ),
5555 op = ComparisonFilter .OP_EQUALS ,
5656 value = AttributeValue (val_double = 123 ),
5757 )
@@ -95,7 +95,7 @@ def test_in_numeric_filter(self):
9595 query , _ = self .resolver .resolve_query ("ai.total_tokens.used:[123,456,789]" )
9696 assert query == TraceItemFilter (
9797 comparison_filter = ComparisonFilter (
98- key = AttributeKey (name = "ai_total_tokens_used" , type = AttributeKey .Type .TYPE_FLOAT ),
98+ key = AttributeKey (name = "ai_total_tokens_used" , type = AttributeKey .Type .TYPE_DOUBLE ),
9999 op = ComparisonFilter .OP_IN ,
100100 value = AttributeValue (val_double_array = DoubleArray (values = [123 , 456 , 789 ])),
101101 )
@@ -105,7 +105,7 @@ def test_greater_than_numeric_filter(self):
105105 query , _ = self .resolver .resolve_query ("ai.total_tokens.used:>123" )
106106 assert query == TraceItemFilter (
107107 comparison_filter = ComparisonFilter (
108- key = AttributeKey (name = "ai_total_tokens_used" , type = AttributeKey .Type .TYPE_FLOAT ),
108+ key = AttributeKey (name = "ai_total_tokens_used" , type = AttributeKey .Type .TYPE_DOUBLE ),
109109 op = ComparisonFilter .OP_GREATER_THAN ,
110110 value = AttributeValue (val_double = 123 ),
111111 )
@@ -281,15 +281,15 @@ def test_simple_string_tag(self):
281281 def test_simple_number_tag (self ):
282282 resolved_column , virtual_context = self .resolver .resolve_column ("tags[foo, number]" )
283283 assert resolved_column .proto_definition == AttributeKey (
284- name = "foo" , type = AttributeKey .Type .TYPE_FLOAT
284+ name = "foo" , type = AttributeKey .Type .TYPE_DOUBLE
285285 )
286286 assert virtual_context is None
287287
288288 def test_sum_function (self ):
289289 resolved_column , virtual_context = self .resolver .resolve_column ("sum(span.self_time)" )
290290 assert resolved_column .proto_definition == AttributeAggregation (
291291 aggregate = Function .FUNCTION_SUM ,
292- key = AttributeKey (name = "sentry.exclusive_time_ms" , type = AttributeKey .Type .TYPE_FLOAT ),
292+ key = AttributeKey (name = "sentry.exclusive_time_ms" , type = AttributeKey .Type .TYPE_DOUBLE ),
293293 label = "sum(span.self_time)" ,
294294 extrapolation_mode = ExtrapolationMode .EXTRAPOLATION_MODE_SAMPLE_WEIGHTED ,
295295 )
@@ -299,7 +299,7 @@ def test_sum_default_argument(self):
299299 resolved_column , virtual_context = self .resolver .resolve_column ("sum()" )
300300 assert resolved_column .proto_definition == AttributeAggregation (
301301 aggregate = Function .FUNCTION_SUM ,
302- key = AttributeKey (name = "sentry.duration_ms" , type = AttributeKey .Type .TYPE_FLOAT ),
302+ key = AttributeKey (name = "sentry.duration_ms" , type = AttributeKey .Type .TYPE_DOUBLE ),
303303 label = "sum()" ,
304304 extrapolation_mode = ExtrapolationMode .EXTRAPOLATION_MODE_SAMPLE_WEIGHTED ,
305305 )
@@ -309,7 +309,7 @@ def test_function_alias(self):
309309 resolved_column , virtual_context = self .resolver .resolve_column ("sum() as test" )
310310 assert resolved_column .proto_definition == AttributeAggregation (
311311 aggregate = Function .FUNCTION_SUM ,
312- key = AttributeKey (name = "sentry.duration_ms" , type = AttributeKey .Type .TYPE_FLOAT ),
312+ key = AttributeKey (name = "sentry.duration_ms" , type = AttributeKey .Type .TYPE_DOUBLE ),
313313 label = "test" ,
314314 extrapolation_mode = ExtrapolationMode .EXTRAPOLATION_MODE_SAMPLE_WEIGHTED ,
315315 )
@@ -319,15 +319,15 @@ def test_count(self):
319319 resolved_column , virtual_context = self .resolver .resolve_column ("count()" )
320320 assert resolved_column .proto_definition == AttributeAggregation (
321321 aggregate = Function .FUNCTION_COUNT ,
322- key = AttributeKey (name = "sentry.duration_ms" , type = AttributeKey .Type .TYPE_FLOAT ),
322+ key = AttributeKey (name = "sentry.duration_ms" , type = AttributeKey .Type .TYPE_DOUBLE ),
323323 label = "count()" ,
324324 extrapolation_mode = ExtrapolationMode .EXTRAPOLATION_MODE_SAMPLE_WEIGHTED ,
325325 )
326326 assert virtual_context is None
327327 resolved_column , virtual_context = self .resolver .resolve_column ("count(span.duration)" )
328328 assert resolved_column .proto_definition == AttributeAggregation (
329329 aggregate = Function .FUNCTION_COUNT ,
330- key = AttributeKey (name = "sentry.duration_ms" , type = AttributeKey .Type .TYPE_FLOAT ),
330+ key = AttributeKey (name = "sentry.duration_ms" , type = AttributeKey .Type .TYPE_DOUBLE ),
331331 label = "count(span.duration)" ,
332332 extrapolation_mode = ExtrapolationMode .EXTRAPOLATION_MODE_SAMPLE_WEIGHTED ,
333333 )
@@ -337,7 +337,7 @@ def test_p50(self):
337337 resolved_column , virtual_context = self .resolver .resolve_column ("p50()" )
338338 assert resolved_column .proto_definition == AttributeAggregation (
339339 aggregate = Function .FUNCTION_P50 ,
340- key = AttributeKey (name = "sentry.duration_ms" , type = AttributeKey .Type .TYPE_FLOAT ),
340+ key = AttributeKey (name = "sentry.duration_ms" , type = AttributeKey .Type .TYPE_DOUBLE ),
341341 label = "p50()" ,
342342 extrapolation_mode = ExtrapolationMode .EXTRAPOLATION_MODE_SAMPLE_WEIGHTED ,
343343 )
0 commit comments