Skip to content

Commit ceed2c9

Browse files
Fixes
1 parent dd0fce0 commit ceed2c9

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

src/utils/variable_manipulations.jl

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -186,12 +186,13 @@ Returns:
186186
- A new `NamedTuple` containing only the specific quantities (e.g., `:q_tot`, `:e_tot`).
187187
"""
188188
@generated function all_specific_gs(gs)
189-
gs_names = Base._nt_names(gs)
190-
relevant_gs_names =
191-
filter(name -> has_prefix(name, ) && name != , gs_names)
192-
specific_gs_names = map(name -> remove_prefix(name, ), relevant_gs_names)
193-
specific_gs_values = map(name -> :(lazy.(specific.(gs.$name, gs.ρ))), relevant_gs_names)
194-
return :(NamedTuple{$specific_gs_names}(($(specific_gs_values...),)))
189+
relevant_names = filter(name -> has_prefix(name, ) && name != , Base._nt_names(eltype(gs)))
190+
specific_names = map(name -> remove_prefix(name, ), relevant_names)
191+
return :(NamedTuple{$specific_names}(
192+
UU.unrolled_map($relevant_names) do name
193+
lazy.(specific.(getproperty(gs, name), gs.ρ))
194+
end
195+
))
195196
end
196197

197198
"""

0 commit comments

Comments
 (0)