From e04f9f92998bb0d80c85cf1bbc340aa786f62cbc Mon Sep 17 00:00:00 2001
From: Felix Koehler <f.koehler@tum.de>
Date: Wed, 12 Jun 2024 14:17:09 +0200
Subject: [PATCH] Fix formatting

---
 tests/test_viz.py | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/tests/test_viz.py b/tests/test_viz.py
index 29d651d..045ac57 100644
--- a/tests/test_viz.py
+++ b/tests/test_viz.py
@@ -7,10 +7,7 @@
 def test_plot_state_1d():
     state = jax.random.normal(
         jax.random.PRNGKey(0),
-        (
-            10,
-            100,
-        ),
+        (10, 100),
     )
 
     fig = ex.viz.plot_state_1d(state)