From 637a6f521435f5802b650c2dd59a4fc369bf1cca Mon Sep 17 00:00:00 2001
From: Anders Kaseorg <andersk@mit.edu>
Date: Tue, 30 Jun 2026 18:41:17 -0700
Subject: [PATCH] fix: restore missing normalization of Dirichlet samples

This was broken by commit a239ebe25006c985115bd1d127804a42413111d4
(#253).  Fixes #344.

Signed-off-by: Anders Kaseorg <andersk@mit.edu>
---
 src/distribution/dirichlet.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/distribution/dirichlet.rs b/src/distribution/dirichlet.rs
index 8191035b..701a8692 100644
--- a/src/distribution/dirichlet.rs
+++ b/src/distribution/dirichlet.rs
@@ -209,7 +209,7 @@ where
                 sum += sample;
                 sample
             }),
-        )
+        ) / sum
     }
 }