Introduce a ParticleSampler type (#86)

* Update smc.jl * Update smc.jl
TuringLang · Oct 23, 2023 · 16cdfa7 · 16cdfa7
1 parent f4eae06
commit 16cdfa7
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/smc.jl b/src/smc.jl
@@ -1,4 +1,6 @@
-struct SMC{R} <: AbstractMCMC.AbstractSampler
+abstract type ParticleSampler <: AbstractMCMC.AbstractSampler end
+
+struct SMC{R} <: ParticleSampler
     nparticles::Int
     resampler::R
 end
@@ -51,7 +53,7 @@ function AbstractMCMC.sample(
     return SMCSample(collect(particles), getweights(particles), logevidence)
 end
 
-struct PG{R} <: AbstractMCMC.AbstractSampler
+struct PG{R} <: ParticleSampler
     """Number of particles."""
     nparticles::Int
     """Resampling algorithm."""
@@ -84,7 +86,7 @@ struct PGSample{T,L}
     logevidence::L
 end
 
-struct PGAS{R} <: AbstractMCMC.AbstractSampler
+struct PGAS{R} <: ParticleSampler
     """Number of particles."""
     nparticles::Int
     """Resampling algorithm."""