fix: initialize HYPER_PARAMETERS in VI subclasses.

sbi-dev · Feb 17, 2025 · 436542d · 436542d
1 parent 18f92b1
commit 436542d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/sbi/samplers/vi/vi_divergence_optimizers.py b/sbi/samplers/vi/vi_divergence_optimizers.py
@@ -513,6 +513,8 @@ def __init__(self, *args, K: int = 8, dreg: bool = False, **kwargs):
         self.loss_name = "iwelbo"
         self.eps = 1e-7
         self.dreg = dreg
+        if not hasattr(self, 'HYPER_PARAMETERS'):
+            self.HYPER_PARAMETERS = []
         self.HYPER_PARAMETERS += ["K", "dreg"]
         if dreg:
             self.stick_the_landing = True
@@ -675,6 +677,8 @@ def __init__(
         self.alpha = alpha
         self.unbiased = unbiased
         super().__init__(*args, **kwargs)
+        if not hasattr(self, 'HYPER_PARAMETERS'):
+            self.HYPER_PARAMETERS = []
         self.HYPER_PARAMETERS += ["alpha", "unbiased", "dreg"]
         self.eps = 1e-5
         self.dreg = dreg