diff --git a/priorConditionedAnnealing/pca.py b/priorConditionedAnnealing/pca.py index d928eae..f17c130 100644 --- a/priorConditionedAnnealing/pca.py +++ b/priorConditionedAnnealing/pca.py @@ -147,7 +147,7 @@ class PCA_Distribution(SB3_Distribution): assert self.skip_conditioning or type(traj) != type(None), 'A past trajectory has to be supplied if conditinoning is performed' pi_mean, pi_std = self.distribution.mean.cpu(), self.distribution.scale.cpu() rho_mean, rho_std = self._conditioning_engine(traj, pi_mean, pi_std) - rho_std *= f_sigma + rho_std = rho_std * f_sigma eta = self._get_rigged(pi_mean, pi_std, rho_mean, rho_std, epsilon)