fix bug in noisy Kuramoto connection

Neuroblox · Jan 16, 2025 · ce333e6 · ce333e6
1 parent 56a18fe
commit ce333e6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/src/blox/connections.jl b/src/blox/connections.jl
@@ -486,8 +486,8 @@ function Connector(
 end
 
 function Connector(
-    blox_src::KuramotoOscillator, 
-    blox_dest::KuramotoOscillator; 
+    blox_src::AbstractKuramotoOscillator, 
+    blox_dest::AbstractKuramotoOscillator; 
     kwargs...
 )
     sys_src = get_namespaced_sys(blox_src)

diff --git a/src/blox/neural_mass.jl b/src/blox/neural_mass.jl
@@ -490,7 +490,9 @@ function kuramoto_oscillator(; name,
     noise ? KuramotoOscillatorNoise(name=name, namespace=namespace, ω=ω, ζ=ζ) : KuramotoOscillator(name=name, namespace=namespace, ω=ω) 
 end
 
-struct KuramotoOscillator <: NeuralMassBlox
+abstract type AbstractKuramotoOscillator <: NeuralMassBlox end
+
+struct KuramotoOscillator <: AbstractKuramotoOscillator
     params
     system
     namespace
@@ -510,7 +512,7 @@ struct KuramotoOscillator <: NeuralMassBlox
     end
 end
 
-struct KuramotoOscillatorNoise <: NeuralMassBlox
+struct KuramotoOscillatorNoise <: AbstractKuramotoOscillator
     params
     system
     namespace