From ac7ea0d477b80e102deb50faff0383c182e77fad Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Mon, 18 Mar 2024 16:01:45 +0200
Subject: [PATCH] Allow filtering task models.input.model field by array of ids

---
 apiserver/database/model/task/task.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/apiserver/database/model/task/task.py b/apiserver/database/model/task/task.py
index d055f8a..5d220a7 100644
--- a/apiserver/database/model/task/task.py
+++ b/apiserver/database/model/task/task.py
@@ -231,11 +231,12 @@ class Task(AttributedDocument):
             "parent",
             "hyperparams.*",
             "execution.queue",
+            "models.input.model",
         ),
         range_fields=("started", "active_duration", "last_metrics.*", "last_iteration"),
         datetime_fields=("status_changed", "last_update"),
         pattern_fields=("name", "comment", "report"),
-        fields=("runtime.*", "models.input.model"),
+        fields=("runtime.*",),
     )
 
     id = StringField(primary_key=True)