Fix MCMC step initialization when not specified by user. Closes #2

ryanjgallagher · Nov 15, 2021 · 90fb289 · 90fb289
1 parent adb17c0
commit 90fb289
Show file tree

Hide file tree

Showing 4 changed files with 1 addition and 5 deletions.
diff --git a/core_periphery_sbm/core_periphery.py b/core_periphery_sbm/core_periphery.py
@@ -1,5 +1,3 @@
-import os
-import sys
 import warnings
 import numpy as np
 import networkx as nx
@@ -246,7 +244,7 @@ def initialize_inference(self, G):
         self.log_posteriors[0] = self.get_log_posterior()
         # Set number of MCMC steps if not set by user
         if self.n_mcmc is None:
-            self.n_mcmc = N * len(self.n_blocks)
+            self.n_mcmc = self.N_nodes * len(self.n_blocks)
 
     def update_block_stats(self, G):
         """

diff --git a/core_periphery_sbm/inference.py b/core_periphery_sbm/inference.py
@@ -1,4 +1,3 @@
-import sys
 import numpy as np
 from scipy.stats import beta
 from scipy.special import loggamma

diff --git a/core_periphery_sbm/model_fit.py b/core_periphery_sbm/model_fit.py
diff --git a/core_periphery_sbm/network_helper.py b/core_periphery_sbm/network_helper.py
@@ -1,5 +1,4 @@
 import numpy as np
-import networkx as nx
 
 def get_max_edges(block_r, block_s, block_ns):
     """