Merge pull request #23 from acxz/size_fix

change size call and use indexing
baggepinnen · May 13, 2021 · da1d6b8 · da1d6b8
2 parents b6d24c1 + 239a60f
commit da1d6b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/backward_pass.jl b/src/backward_pass.jl
@@ -178,7 +178,7 @@ end
 
 function back_pass(cx,cu,cxx::AbstractArray{T,3},cxu,cuu,fx::AbstractArray{T,3},fu,λ,regType,lims,x,u) where T # quadratic timeVariant cost, linear time variant dynamics
     m          = size(u,1)
-    n,N        = size(fx,1,3)
+    n,N        = size(fx, 1), size(fx, 3)
 
     @assert size(cx) == (n, N)
     @assert size(cu) == (m, N)