force kw arguments

e3nn · Jan 5, 2024 · 91fac35 · 91fac35
1 parent 1387c7d
commit 91fac35
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/e3nn_jax/_src/linear_equinox.py b/e3nn_jax/_src/linear_equinox.py
@@ -122,11 +122,11 @@ class Linear(eqx.Module):
 
     def __init__(
         self,
+        *,
         irreps_out: e3nn.Irreps,
         irreps_in: e3nn.Irreps,
         channel_out: Optional[int] = None,
         channel_in: Optional[int] = None,
-        *,
         biases: bool = False,
         path_normalization: Optional[Union[str, float]] = None,
         gradient_normalization: Optional[Union[str, float]] = None,