dodox/PriorConditionedAnnealing
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Typo

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-05-21 17:26:21 +02:00
parent 852e1817f7
commit 4ba7db831f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
priorConditionedAnnealing/pca.py
View File

@ -143,7 +143,7 @@ class PCA_Distribution(SB3_Distribution):
""" """
if deterministic: if deterministic:
return self.mode() 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: def sample(self, traj: th.Tensor, f_sigma: int = 1, epsilon=None) -> th.Tensor:
pi_mean, pi_std = self.distribution.mean, self.distribution.scale pi_mean, pi_std = self.distribution.mean, self.distribution.scale