Update thermox/prob.py

Co-authored-by: SamDuffield <34280297+SamDuffield@users.noreply.github.com>
normal-computing · May 9, 2024 · 1b9aeda · 1b9aeda
1 parent 74db602
commit 1b9aeda
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)