Remove 10 metrics limit in multi-task plot comparison

This commit is contained in:
allegroai 2023-07-26 18:55:49 +03:00
parent 6c5f966ed4
commit f84cc0a2cb

View File

@ -881,6 +881,7 @@ class EventBLL(object):
max_metrics, max_variants = get_max_metric_and_variant_counts(
query=query, **search_args,
)
max_variants = int(max_variants // 2)
es_req = {
"size": 0,
"query": query,
@ -1004,14 +1005,16 @@ class EventBLL(object):
if metrics:
must.append(get_metric_variants_condition(metrics))
max_tasks = min(len(task_ids), 1000)
max_metrics = 10_000 // (max_tasks * iters)
es_req: dict = {
"size": 0,
"aggs": {
"tasks": {
"terms": {"field": "task"},
"terms": {"field": "task", "size": max_tasks},
"aggs": {
"metrics": {
"terms": {"field": "metric"},
"terms": {"field": "metric", "size": max_metrics},
"aggs": {
"iters": {
"terms": {
@ -1066,11 +1069,12 @@ class EventBLL(object):
if metrics:
must.append(get_metric_variants_condition(metrics))
max_tasks = min(len(task_ids), 1000)
es_req: dict = {
"size": 0,
"aggs": {
"tasks": {
"terms": {"field": "task"},
"terms": {"field": "task", "size": max_tasks},
"aggs": {
"iters": {
"terms": {