fix: use projects resolver to check for data connections

kili-technology · Oct 5, 2023 · 0b434df · 0b434df
1 parent 662437d
commit 0b434df
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/fakes/fake_kili.py b/tests/fakes/fake_kili.py
@@ -93,6 +93,7 @@ def mocked_ProjectQuery(where, _fields, _options):
                 "description": "This is a test project",
                 "jsonInterface": json_interface,
                 "inputType": "IMAGE",
+                "dataConnections": None,
             }
         ]
     elif project_id == "object_detection_video_project":
@@ -122,6 +123,7 @@ def mocked_ProjectQuery(where, _fields, _options):
                 "description": "This is a test project",
                 "jsonInterface": json_interface,
                 "inputType": "VIDEO",
+                "dataConnections": None,
             }
         ]
     elif project_id == "text_classification":
@@ -155,6 +157,7 @@ def mocked_ProjectQuery(where, _fields, _options):
                 "description": "This is a TC test project",
                 "jsonInterface": json_interface,
                 "inputType": "TEXT",
+                "dataConnections": None,
             }
         ]
     elif project_id == "semantic_segmentation":
@@ -198,6 +201,7 @@ def mocked_ProjectQuery(where, _fields, _options):
                 "description": "This is a semantic segmentation test project",
                 "jsonInterface": json_interface,
                 "inputType": "IMAGE",
+                "dataConnections": None,
             }
         ]
     else: