make sure pullback is in [0,1]

benjione · Nov 4, 2023 · da0a9a5 · da0a9a5
1 parent ff97d69
commit da0a9a5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/Samplers/PSDModelSampler.jl b/src/Samplers/PSDModelSampler.jl
@@ -83,6 +83,7 @@ function _pullback_first_n(sampler::PSDModelSampler{d, T},
     for k=1:n
         u[sampler.variable_ordering[k]] = f(k)(x[k])
     end
+    u = map(x->x ≤ 0 ? (x ≥ one(T) ? x : one(T)) : zero(T) : x, u)
     return u
 end