validmind · juanmleng · Jan 15, 2025 · Jan 14, 2025 · Jan 14, 2025 · Jan 15, 2025
diff --git a/notebooks/how_to/log_metrics_over_time.ipynb b/notebooks/how_to/log_metrics_over_time.ipynb
diff --git a/notebooks/images/add_metric_over_time_block.png b/notebooks/images/add_metric_over_time_block.png
diff --git a/notebooks/images/log_metric_accuracy.png b/notebooks/images/log_metric_accuracy.png
diff --git a/notebooks/images/log_metric_auc_1.png b/notebooks/images/log_metric_auc_1.png
diff --git a/notebooks/images/log_metric_auc_2.png b/notebooks/images/log_metric_auc_2.png
diff --git a/notebooks/images/log_metric_auc_3.png b/notebooks/images/log_metric_auc_3.png
diff --git a/notebooks/images/log_metric_auc_4.png b/notebooks/images/log_metric_auc_4.png
diff --git a/notebooks/images/log_metric_f1.png b/notebooks/images/log_metric_f1.png
diff --git a/notebooks/images/log_metric_precision.png b/notebooks/images/log_metric_precision.png
diff --git a/notebooks/images/log_metric_recall.png b/notebooks/images/log_metric_recall.png
diff --git a/pyproject.toml b/pyproject.toml
@@ -10,7 +10,7 @@ description = "ValidMind Library"
 license = "Commercial License"
 name = "validmind"
 readme = "README.pypi.md"
-version = "2.7.6"
+version = "2.7.7"
 
 [tool.poetry.dependencies]
 aiohttp = {extras = ["speedups"], version = "*"}

diff --git a/validmind/__version__.py b/validmind/__version__.py
@@ -1 +1 @@
-__version__ = "2.7.6"
+__version__ = "2.7.7"
diff --git a/validmind/api_client.py b/validmind/api_client.py
@@ -407,6 +407,7 @@ async def alog_metric(
     inputs: Optional[List[str]] = None,
     params: Optional[Dict[str, Any]] = None,
     recorded_at: Optional[str] = None,
+    thresholds: Optional[Dict[str, Any]] = None,
 ):
     """See log_metric for details"""
     if not key or not isinstance(key, str):
@@ -421,6 +422,9 @@ async def alog_metric(
         except (ValueError, TypeError):
             raise ValueError("`value` must be a scalar (int or float)")
 
+    if thresholds is not None and not isinstance(thresholds, dict):
+        raise ValueError("`thresholds` must be a dictionary or None")
+
     try:
         return await _post(
             "log_unit_metric",
@@ -431,6 +435,7 @@ async def alog_metric(
                     "inputs": inputs or [],
                     "params": params or {},
                     "recorded_at": recorded_at,
+                    "thresholds": thresholds or {},
                 },
                 cls=NumpyEncoder,
                 allow_nan=False,
@@ -447,6 +452,7 @@ def log_metric(
     inputs: Optional[List[str]] = None,
     params: Optional[Dict[str, Any]] = None,
     recorded_at: Optional[str] = None,
+    thresholds: Optional[Dict[str, Any]] = None,
 ):
     """Logs a unit metric
 
@@ -463,8 +469,9 @@ def log_metric(
         params (dict, optional): Dictionary of parameters used to compute the metric.
         recorded_at (str, optional): The timestamp of the metric. Server will use
             current time if not provided.
+        thresholds (dict, optional): Dictionary of thresholds for the metric.
     """
-    run_async(alog_metric, key, value, inputs, params, recorded_at)
+    run_async(alog_metric, key, value, inputs, params, recorded_at, thresholds)
 
 
 def get_ai_key() -> Dict[str, Any]: