Merge updated files

2022-02-17 22:41:55 +01:00
parent b78f95bab5
commit 5bd8b42d9f
83 changed files with 1439 additions and 1123 deletions
--- a/src/core/reparam_module.py
+++ b/src/core/reparam_module.py
@@ -160,3 +160,6 @@ class ReparamPolicy(ReparamModule):
    
    def predict(self, *args, **kwargs):
        return self.module.predict(*args, **kwargs)
+    
+    def unsafe_probability_mass(self, *args, **kwargs):
+        return self.module.unsafe_probability_mass(*args, **kwargs)