From f084f6b9e797a3ced23c83932220c5356988787b Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Tue, 5 Jan 2021 17:38:50 +0200
Subject: [PATCH] Fix get_hyperparams and unique_metric_params handling of
 public tasks

---
 apiserver/bll/task/task_bll.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/apiserver/bll/task/task_bll.py b/apiserver/bll/task/task_bll.py
index 48af7ff..d3d69e3 100644
--- a/apiserver/bll/task/task_bll.py
+++ b/apiserver/bll/task/task_bll.py
@@ -289,7 +289,7 @@ class TaskBLL(object):
         pipeline = [
             {
                 "$match": dict(
-                    company=company_id,
+                    company={"$in": [None, "", company_id]},
                     **({"project": {"$in": project_ids}} if project_ids else {}),
                 )
             },
@@ -558,7 +558,7 @@ class TaskBLL(object):
         pipeline = [
             {
                 "$match": {
-                    "company": company_id,
+                    "company": {"$in": [None, "", company_id]},
                     "hyperparams": {"$exists": True, "$gt": {}},
                     **({"project": {"$in": project_ids}} if project_ids else {}),
                 }