Apply suggestions from code review

Co-authored-by: David Widmann <devmotion@users.noreply.github.com>
TuringLang · Aug 21, 2020 · de86456 · de86456
1 parent 01250ce
commit de86456
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/bijectors/corr.jl b/src/bijectors/corr.jl
@@ -46,8 +46,7 @@ end
 
 
 function _inv_link_chol_lkj(y)
-    @assert size(y, 1) == size(y, 2)
-    K = size(y, 1)
+    K = LinearAlgebra.checksquare(y)
 
     z = tanh.(y)
     w = similar(z)
@@ -76,8 +75,7 @@ function _inv_link_chol_lkj(y)
 end
 
 function _link_chol_lkj(w)
-    @assert size(w, 1) == size(w, 2)
-    K = size(w, 1)
+    K = LinearAlgebra.checksquare(w)
 
     z = zero(w)