fix: Optimize fetch_metric_buckets for <1k series (#108)

pitercl · web-flow · commit cfd13b327149 · 2025-10-08T16:04:15.000Z
diff --git a/src/neptune_query/internal/composition/fetch_metric_buckets.py b/src/neptune_query/internal/composition/fetch_metric_buckets.py
@@ -189,8 +189,8 @@ def go_fetch_sys_attrs() -> Generator[list[identifiers.SysId], None, None]:
         executor=executor,
     )
 
-    # if len(run_attribute_definitions) <= MAX_SERIES_PER_REQUEST:
-    #     return fetch_in_chunks(x_range=None, bucket_limit=limit), sys_id_label_mapping
+    if len(run_attribute_definitions) <= MAX_SERIES_PER_REQUEST:
+        return fetch_in_chunks(x_range=None, bucket_limit=limit), sys_id_label_mapping
 
     global_x_range = _compute_global_x_range(fetch_in_chunks=fetch_in_chunks)
     if global_x_range is None:

Original file line number	Diff line number	Diff line change
`@@ -189,8 +189,8 @@ def go_fetch_sys_attrs() -> Generator[list[identifiers.SysId], None, None]:`
`189`	`189`	`executor=executor,`
`190`	`190`	`)`
`191`	`191`
`192`		`- # if len(run_attribute_definitions) <= MAX_SERIES_PER_REQUEST:`
`193`		`- # return fetch_in_chunks(x_range=None, bucket_limit=limit), sys_id_label_mapping`
	`192`	`+ if len(run_attribute_definitions) <= MAX_SERIES_PER_REQUEST:`
	`193`	`+ return fetch_in_chunks(x_range=None, bucket_limit=limit), sys_id_label_mapping`
`194`	`194`
`195`	`195`	`global_x_range = _compute_global_x_range(fetch_in_chunks=fetch_in_chunks)`
`196`	`196`	`if global_x_range is None:`