make basis modes as static attribute

PlasmaControl · Feb 12, 2025 · 6365a46 · 6365a46
1 parent 72d8970
commit 6365a46
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/desc/basis.py b/desc/basis.py
@@ -34,6 +34,7 @@ class _Basis(IOAble, ABC):
         "_sym",
         "_spectral_indexing",
     ]
+    _static_attrs = ["_modes"]
 
     def __init__(self):
         self._enforce_symmetry()

diff --git a/desc/equilibrium/equilibrium.py b/desc/equilibrium/equilibrium.py
@@ -163,6 +163,7 @@ class Equilibrium(IOAble, Optimizable):
         "_M_grid",
         "_N_grid",
     ]
+    _static_attrs = ["_R_basis", "_Z_basis", "_L_basis"]
 
     @execute_on_cpu
     def __init__(

diff --git a/desc/geometry/surface.py b/desc/geometry/surface.py
@@ -65,6 +65,7 @@ class FourierRZToroidalSurface(Surface):
         "_NFP",
         "_rho",
     ]
+    _static_attrs = ["_R_basis", "_Z_basis"]
 
     @execute_on_cpu
     def __init__(
-Original file line number
+Diff line change
@@ Expand Up / @@ -65,6 +65,7 @@ class FourierRZToroidalSurface(Surface): @@
             "_NFP",
             "_rho",
         ]
+        _static_attrs = ["_R_basis", "_Z_basis"]
         @execute_on_cpu
         def __init__(
@@ Expand Down @@