Formatting

Co-authored-by: vfdev <[email protected]>

Author: jorisSchaller

tests/pooling_test.py

@@ -1,5 +1,5 @@
 import unittest
-from flax.pooling import pool,avg_pool, max_pool, min_pool
+from flax.pooling import pool, avg_pool, max_pool, min_pool
 import numpy as np
 import jax.numpy as jnp
 from absl.testing import absltest, parameterized