Minor things

Author: jagot

src/dyson_expansions.jl

@@ -394,7 +394,8 @@ set_momentum!(𝐩s::AbstractVector{<:SVector{3}}, 𝐩ₛₜ::SVector{3}, i) =
 set_momentum!(𝐩s::AbstractVector{<:SVector{3}}, 𝐩ₛₜ::T, i) where {T<:Number} =
     setindex!(𝐩s, SVector{3,T}(zero(T), zero(T), 𝐩ₛₜ), i)
 
-function evaluate_momenta!(𝐩s, prefactors, system, unique_momenta, indeterminate_momenta, i; ϵ=1e-2*√(eps(eltype(system.t))))
+function evaluate_momenta!(𝐩s, prefactors, system, unique_momenta, indeterminate_momenta, i;
+                           ϵ=1e-2*√(eps(eltype(system.t))))
     for idm in indeterminate_momenta
         uidm = unique_momenta[idm]
         a,b = i[uidm[1]],i[uidm[2]]

src/momentum_grid.jl

@@ -1,11 +1,15 @@
-function momentum_grid(kmin, kmax, nk, nθ; spacing=:momentum)
-    kmag = if spacing == :momentum
+function momentum_grid(kmin, kmax, nk; spacing=:momentum)
+    if spacing == :momentum
         range(kmin, stop=kmax, length=nk)
     elseif spacing == :energy
         .√(2range(kmin^2/2, stop=kmax^2/2, length=nk))
     else
         throw(ArgumentError("Unknown spacing $(spacing)"))
     end
+end
+
+function momentum_grid(kmin, kmax, nk, nθ, nϕ=1; kwargs...)
+    kmag = momentum_grid(kmin, kmax, nk; kwargs...)
     nθ == 1 && return (kmag,kmag,nothing)
 
     x = range(0, stop=1, length=nθ)

src/telescope_iterators.jl

@@ -4,7 +4,7 @@
 Iterator that is formally equivalent to
 ```julia
 for i = 1:length(v), j = max(1,i-memory):i-1, k = max(1,j-memory):j-1, ...
-    e = v[i,j,k,...]
+    e = v[[i,j,k,...]]
 end
 ```
 for `n` loop variables.

src/threading.jl

@@ -22,8 +22,8 @@ function threaded_range_loop(fun::Function, n::Integer)
     end
 end
 
-threaded_range_loop(fun::Function, v::AbstractVector) =
+threaded_range_loop(fun::Function, v::AbstractArray) =
     threaded_range_loop(i -> fun(v[i]), length(v))
 
-threaded_enumerate_range_loop(fun::Function, v::AbstractVector) =
+threaded_enumerate_range_loop(fun::Function, v::AbstractArray) =
     threaded_range_loop(i -> fun(i,v[i]), length(v))