Correct typo.

astro-informatics · May 10, 2024 · ac94d94 · ac94d94
1 parent da4b31f
commit ac94d94
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/harmonic/flows.py b/harmonic/flows.py
@@ -210,7 +210,7 @@ class RQSpline(nn.Module):
     num_bins: int
     spline_range: Sequence[float] = (-10.0, 10.0)
     multimodal_base: bool = False
-    base_centers: Sequence[jnp.ndarray]] = None
+    base_centers: Sequence[jnp.ndarray] = None
 
     def setup(self):
         conditioner = []