From ab1b269af9ef6a960e1188f0d5e0c5a37fb21206 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Fri, 1 Jul 2022 11:52:14 +0200 Subject: [PATCH] Allow checking whether a dist is contextual --- metastable_baselines/misc/distTools.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/metastable_baselines/misc/distTools.py b/metastable_baselines/misc/distTools.py index 633992e..0e7d197 100644 --- a/metastable_baselines/misc/distTools.py +++ b/metastable_baselines/misc/distTools.py @@ -52,6 +52,11 @@ def has_diag_cov(p, numerical_check=True): return th.equal(cov - th.diag_embed(th.diagonal(cov, dim1=-2, dim2=-1), th.zeros_like(cov))) +def is_contextual(p): + # TODO: Implement for UniveralGaussianDist + return False + + def get_diag_cov_vec(p, check_diag=True, numerical_check=True): if check_diag and not has_diag_cov(p): raise Exception('Cannot reduce cov-mat to diag-vec: Is not diagonal')