Bump version to v2.5.0-rc3

polyaxon · Oct 4, 2024 · b712e30 · b712e30
1 parent 9d033fb
commit b712e30
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 11 deletions.
diff --git a/haupt/haupt/db/managers/stats.py b/haupt/haupt/db/managers/stats.py
@@ -216,28 +216,30 @@ def filter_queryset(cls, queryset, request, view):
         return queryset
 
 
-def annotate_statuses(queryset):
-    return queryset.annotate(
-        count=Count(
-            "runs",
-            distinct=True,
-        ),
-        running=Count(
+def annotate_statuses(queryset, include_total=False):
+    agg = {
+        "running": Count(
             "runs",
             filter=Q(runs__status__in=LifeCycle.RUNNING_VALUES),
             distinct=True,
         ),
-        pending=Count(
+        "pending": Count(
             "runs",
             filter=Q(runs__status__in=LifeCycle.ALL_PENDING_VALUES),
             distinct=True,
         ),
-        warning=Count(
+        "warning": Count(
             "runs",
             filter=Q(runs__status__in=LifeCycle.ALL_WARNING_VALUES),
             distinct=True,
         ),
-    )
+    }
+    if include_total:
+        agg["count"] = Count(
+            "runs",
+            distinct=True,
+        )
+    return queryset.annotate(**agg)
 
 
 def annotate_quota(queryset):

diff --git a/haupt/haupt/pkg.py b/haupt/haupt/pkg.py
@@ -1,5 +1,5 @@
 NAME = "haupt"
-VERSION = "2.4.0"
+VERSION = "2.5.0-rc3"
 DESC = (
     "Lineage metadata API, artifacts streams, sandbox, ML-API, and spaces for Polyaxon."
 )