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

Fixing dependency problems

Browse Source
This commit is contained in:
Dominik Moritz Roth 2022-09-03 11:40:14 +02:00
parent 6c7fc37116
commit c7ca326345
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
metastable_projections/misc/distTools.py
View File

@ -1,8 +1,15 @@
from typing import Any, Dict, List, Optional, Tuple, Union
import torch as th
from stable_baselines3.common.distributions import Distribution as SB3_Distribution
from ..distributions import UniversalGaussianDistribution, AnyDistribution
class UniversalGaussianDistribution(SB3_Distribution):
pass
AnyDistribution = Union[SB3_Distribution, UniversalGaussianDistribution]
def get_mean_and_chol(p: AnyDistribution, expand=False):