Fix formatting

Ceyron · Jun 12, 2024 · e04f9f9 · e04f9f9
1 parent 452a4d0
commit e04f9f9
Showing 1 changed file with 1 addition and 4 deletions.
diff --git 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)