We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 30c0d66 commit d2f4814Copy full SHA for d2f4814
torax/_src/geometry/geometry.py
@@ -340,11 +340,7 @@ def g1_over_vpr2_face(self) -> jax.Array:
340
@property
341
def gm9(self) -> jax.Array:
342
r"""<1/R> on cell grid [:math:`\mathrm{m}^{-1}`]."""
343
- bulk = 2 * jnp.pi * self.spr[..., 1:] / self.vpr[..., 1:]
344
- first_element = 1 / self.R_major
345
- return jnp.concatenate(
346
- [jnp.expand_dims(first_element, axis=-1), bulk], axis=-1
347
- )
+ return 2 * jnp.pi * self.spr / self.vpr
348
349
350
def gm9_face(self) -> jax.Array:
0 commit comments