don't forget Bernoulli prior

vandenman · vandenman · commit 7924be1ff55b · 2025-09-25T16:17:24.000+02:00
diff --git a/src/common_helpers.h b/src/common_helpers.h
@@ -41,7 +41,7 @@ inline UpdateMethod update_method_from_string(const std::string& update_method)
   throw std::invalid_argument("Invalid update_method: " + update_method);
 }
 
-enum EdgePrior { Stochastic_Block, Beta_Bernoulli };
+enum EdgePrior { Stochastic_Block, Beta_Bernoulli, Bernoulli };
 
 inline EdgePrior edge_prior_from_string(const std::string& edge_prior) {
   if (edge_prior == "stochastic-block")
@@ -50,5 +50,8 @@ inline EdgePrior edge_prior_from_string(const std::string& edge_prior) {
   if (edge_prior == "Beta-Bernoulli")
     return Beta_Bernoulli;
 
+  if (edge_prior == "Bernoulli")
+    return Bernoulli;
+
   throw std::invalid_argument("Invalid edge_prior: " + edge_prior);
 }

Original file line number	Diff line number	Diff line change
`@@ -41,7 +41,7 @@ inline UpdateMethod update_method_from_string(const std::string& update_method)`
`41`	`41`	`throw std::invalid_argument("Invalid update_method: " + update_method);`
`42`	`42`	`}`
`43`	`43`
`44`		`-enum EdgePrior { Stochastic_Block, Beta_Bernoulli };`
	`44`	`+enum EdgePrior { Stochastic_Block, Beta_Bernoulli, Bernoulli };`
`45`	`45`
`46`	`46`	`inline EdgePrior edge_prior_from_string(const std::string& edge_prior) {`
`47`	`47`	`if (edge_prior == "stochastic-block")`
`@@ -50,5 +50,8 @@ inline EdgePrior edge_prior_from_string(const std::string& edge_prior) {`
`50`	`50`	`if (edge_prior == "Beta-Bernoulli")`
`51`	`51`	`return Beta_Bernoulli;`
`52`	`52`
	`53`	`+ if (edge_prior == "Bernoulli")`
	`54`	`+ return Bernoulli;`
	`55`	`+`
`53`	`56`	`throw std::invalid_argument("Invalid edge_prior: " + edge_prior);`
`54`	`57`	`}`