File tree Expand file tree Collapse file tree 1 file changed +7
-6
lines changed Expand file tree Collapse file tree 1 file changed +7
-6
lines changed Original file line number Diff line number Diff line change @@ -186,12 +186,13 @@ Returns:
186
186
- A new `NamedTuple` containing only the specific quantities (e.g., `:q_tot`, `:e_tot`).
187
187
"""
188
188
@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
+ ))
195
196
end
196
197
197
198
"""
You can’t perform that action at this time.
0 commit comments