add more explicit type constraints

MurrellGroup · Mar 4, 2024 · 02889d5 · 02889d5
1 parent 7bb4af6
commit 02889d5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/MessagePassingIPA.jl b/src/MessagePassingIPA.jl
@@ -232,10 +232,10 @@ sumdrop(x; dims) = dropdims(sum(x; dims); dims)
 # ---------------------------
 
 struct GeometricVectorPerceptron
-    W_h
-    W_μ
+    W_h::AbstractMatrix
+    W_μ::AbstractMatrix
     scalar::Dense
-    vσ
+    vσ::Function
 end
 
 Flux.@functor GeometricVectorPerceptron