Skip to content

Commit d2f4814

Browse files
committed
Fix gm9 cell calculation
1 parent 30c0d66 commit d2f4814

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

torax/_src/geometry/geometry.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -340,11 +340,7 @@ def g1_over_vpr2_face(self) -> jax.Array:
340340
@property
341341
def gm9(self) -> jax.Array:
342342
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-
)
343+
return 2 * jnp.pi * self.spr / self.vpr
348344

349345
@property
350346
def gm9_face(self) -> jax.Array:

0 commit comments

Comments
 (0)