Set multiple lenses successively using Kaleido.jl

See jw3126/Setfield.jl#74
MineralsCloud · Aug 13, 2019 · 6869bb4 · 6869bb4
1 parent 28a2232
commit 6869bb4
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/StructureOptimization/StructureOptimization.jl b/src/StructureOptimization/StructureOptimization.jl
@@ -13,6 +13,8 @@ module StructureOptimization
 
 using LinearAlgebra
 
+using Kaleido: @batchlens
+
 using IntervalArithmetic
 using IntervalRootFinding
 using EquationsOfState
@@ -34,9 +36,11 @@ function update_alat(pw::PWscfInput, eos::EquationOfState, pressure::Real)
     volume = find_volume(PressureTarget, eos, pressure, 0..1000, Newton).interval.lo
     alat = cbrt(volume / det(pw.cell_parameters.data))
 
-    alat_lens = @lens _.system.celldm[1]
-    press_lens = @lens _.cell.press
-    set(set(pw, alat_lens, alat), press_lens, pressure)
+    lenses = @batchlens begin
+        _.system.celldm[1]
+        _.cell.press
+    end
+    set(pw, lenses, (alat, pressure))
 end
 
 function generate_input(