From 548674ecc7c1706a28adef51d2e9d15c9514822b Mon Sep 17 00:00:00 2001 From: unalmis Date: Thu, 19 Dec 2024 05:45:40 -0500 Subject: [PATCH] . --- desc/compute/_fast_ion.py | 7 +++---- desc/integrals/bounce_integral.py | 2 +- 2 files changed, 4 insertions(+), 5 deletions(-) diff --git a/desc/compute/_fast_ion.py b/desc/compute/_fast_ion.py index b24c24fcb..43f388c88 100644 --- a/desc/compute/_fast_ion.py +++ b/desc/compute/_fast_ion.py @@ -130,10 +130,9 @@ def _Gamma_c(params, transforms, profiles, data, **kwargs): num_well = kwargs.get("num_well", Y_B * num_transit) batch_size = kwargs.get("batch_size", None) spline = kwargs.get("spline", True) - if "fieldline_quad" in kwargs: - fieldline_quad = kwargs["fieldline_quad"] - else: - fieldline_quad = leggauss(Y_B // 2) + fieldline_quad = ( + kwargs["fieldline_quad"] if "fieldline_quad" in kwargs else leggauss(Y_B // 2) + ) quad = ( kwargs["quad"] if "quad" in kwargs diff --git a/desc/integrals/bounce_integral.py b/desc/integrals/bounce_integral.py index b1d130529..cc7bc0dbf 100644 --- a/desc/integrals/bounce_integral.py +++ b/desc/integrals/bounce_integral.py @@ -70,7 +70,7 @@ def _swap_pl(f): class Bounce2D(Bounce): - """Computes bounce integrals using two-dimensional pseudo-spectral methods. + """Computes bounce integrals using pseudo-spectral methods. The bounce integral is defined as ∫ f(ρ,α,λ,ℓ) dℓ where