From 169c4bbada5853af5e3566ada454cd334b244db0 Mon Sep 17 00:00:00 2001
From: dpsimpson <dp.simpson@gmail.com>
Date: Thu, 17 Oct 2024 14:42:00 +0100
Subject: [PATCH] Black formatted

---
 tests/statistical/test_generate.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/statistical/test_generate.py b/tests/statistical/test_generate.py
index c258acdf..aa1e1e40 100644
--- a/tests/statistical/test_generate.py
+++ b/tests/statistical/test_generate.py
@@ -24,7 +24,7 @@ class NextToken:
         probs: dict[int, List[float]] = {
             1: [0.2, 0.5, 0.3],
             2: [0.3, 0.4, 0.3],
-            3: [0, 0, 0]
+            3: [0, 0, 0],
         }
         p0: List[float] = [0.2, 0.8, 0.0]
         states: List[int] = [1, 2, 3]