Allow requesting plots and iter_histograms for all variants

This commit is contained in:
allegroai 2021-07-25 14:30:38 +03:00
parent 9069cfe1da
commit 6705aff56f
2 changed files with 3 additions and 5 deletions
apiserver
apimodels
bll/task

View File

@ -20,9 +20,7 @@ class HistogramRequestBase(Base):
class MetricVariants(Base):
metric: str = StringField(required=True)
variants: Sequence[str] = ListField(
items_types=str, validators=Length(minimum_value=1)
)
variants: Sequence[str] = ListField(items_types=str)
class ScalarMetricsIterHistogramRequest(HistogramRequestBase):

View File

@ -130,14 +130,14 @@ def collect_debug_image_urls(company: str, task: str) -> Set[str]:
if not metrics:
return set()
task_metrics = {task: set(metrics)}
task_metrics = {task: {m: [] for m in metrics}}
scroll_id = None
urls = set()
while True:
res = event_bll.debug_images_iterator.get_task_events(
company_id=company,
task_metrics=task_metrics,
iter_count=100,
iter_count=10,
state_id=scroll_id,
)
if not res.metric_events or not any(