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