return only pymc key-value pair

bambinos · Jul 4, 2024 · 70532fc · 70532fc
1 parent 2e824d8
commit 70532fc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/bambi/backend/inference_methods.py b/bambi/backend/inference_methods.py
@@ -72,6 +72,8 @@ def get_kwargs(self, method):
 
     @property
     def names(self):
+        if self.bayeux_model is None:
+            return {"pymc": self.pymc_methods}
         return {"pymc": self.pymc_methods, "bayeux": self.bayeux_methods}