From 4ba7db831f1ed58dd3ed3ff068f290f40a6ecd1c Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Sun, 21 May 2023 17:26:21 +0200 Subject: [PATCH] Typo --- priorConditionedAnnealing/pca.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/priorConditionedAnnealing/pca.py b/priorConditionedAnnealing/pca.py index e37e231..b71efaa 100644 --- a/priorConditionedAnnealing/pca.py +++ b/priorConditionedAnnealing/pca.py @@ -143,7 +143,7 @@ class PCA_Distribution(SB3_Distribution): """ if deterministic: return self.mode() - return self.sample(trajectory=trajectory) + return self.sample(traj=trajectory) def sample(self, traj: th.Tensor, f_sigma: int = 1, epsilon=None) -> th.Tensor: pi_mean, pi_std = self.distribution.mean, self.distribution.scale