Merge pull request #101 from AI4Bharat/dev-s

AI4Bharat · Aug 21, 2024 · 4f027c2 · 4f027c2
2 parents 08f88cc + c698ba2
commit 4f027c2
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/backend/workspaces/views.py b/backend/workspaces/views.py
@@ -1665,14 +1665,14 @@ def cumulative_tasks_count_all(self, request, pk=None):
             metainfo = request.query_params["metainfo"]
             if metainfo == "true" or metainfo == "True":
                 metainfo = True
-        project_types = [
-            "ContextualTranslationEditing",
-            "ContextualSentenceVerification",
-            "SemanticTextualSimilarity_Scale5",
-            "AudioTranscriptionEditing",
-            "AudioTranscription",
-            "AudioSegmentation",
-        ]
+        if "project_type_filter" in dict(request.query_params):
+            project_types = [request.query_params["project_type_filter"]]
+        else:
+            project_types = [
+                "InstructionDrivenChat",
+                "ModelInteractionEvaluation",
+                "ModelOutputEvaluation",
+            ]
         if "project_type" in dict(request.query_params):
             project_type = request.query_params["project_type"]
             project_types = [project_type]