From 1b9aeda86b3667ae66cdedfb0138397725cf8dd5 Mon Sep 17 00:00:00 2001
From: KaelanDt <kaelandonatella@gmail.com>
Date: Thu, 9 May 2024 12:39:25 +0100
Subject: [PATCH] Update thermox/prob.py

Co-authored-by: SamDuffield <34280297+SamDuffield@users.noreply.github.com>
---
 thermox/prob.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/thermox/prob.py b/thermox/prob.py
index 335f886..b710402 100644
--- a/thermox/prob.py
+++ b/thermox/prob.py
@@ -33,7 +33,7 @@ def log_prob_identity_diffusion(
         A: drift matrix (Array or thermox.ProcessedDriftMatrix).
         b: drift displacement vector.
     Returns:
-        log probability of given xs.
+        Scalar log probability of given xs.
     """
     if isinstance(A, Array):
         A = preprocess_drift_matrix(A)