feat: cache of metric values for faster charts (#171)

zeno-ml · Sep 13, 2023 · 4f5a28b · 4f5a28b
1 parent 230b016
commit 4f5a28b
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/backend/zeno_backend/classes/metric.py b/backend/zeno_backend/classes/metric.py
@@ -18,6 +18,14 @@ class Metric(CamelModel):
     type: str
     columns: list[str]
 
+    def __hash__(self):
+        """Hash representation of metric, based on id since guaranteed unique.
+
+        Returns:
+            int: the hash of the metric.
+        """
+        return self.id
+
 
 class MetricKey(CamelModel):
     """Specification of metric keys in zeno.

diff --git a/backend/zeno_backend/processing/metrics/map.py b/backend/zeno_backend/processing/metrics/map.py
@@ -1,5 +1,7 @@
 """Map metric names to metric functions."""
 
+from functools import lru_cache
+
 from psycopg import sql
 
 from zeno_backend.classes.base import GroupMetric
@@ -51,6 +53,7 @@ def count(
         )
 
 
+@lru_cache(maxsize=4096)
 def metric_map(
     metric: Metric | None,
     project: str,