dodox/metastable-baselines
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix: Dependency

Browse Source
This commit is contained in:
Dominik Moritz Roth 2022-09-03 23:41:06 +02:00
parent d3fa3cc997
commit 2c14edd3b0
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
metastable_baselines/distributions/distributions.py
View File

@ -7,8 +7,6 @@ from torch import nn
from torch.distributions import Normal, Independent, MultivariateNormal
from math import pi
from misc import givens
from stable_baselines3.common.preprocessing import get_action_dim
from stable_baselines3.common.distributions import sum_independent_dims
@ -23,6 +21,7 @@ from stable_baselines3.common.distributions import DiagGaussianDistribution
from ..misc.tensor_ops import fill_triangular
from ..misc.tanhBijector import TanhBijector
from ..misc import givens
class Strength(Enum):