🐛 warning message fix

GiulioRossetti · May 30, 2019 · ec67855 · ec67855 · ggrrll · May 31, 2019
1 parent afa1767
commit ec67855
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/ndlib/models/DiffusionModel.py b/ndlib/models/DiffusionModel.py
@@ -158,7 +158,9 @@ def set_initial_status(self, configuration):
             if 'fraction_infected' in self.params['model']:
                 number_of_initial_infected = len(self.graph.nodes()) * float(self.params['model']['fraction_infected'])
                 if number_of_initial_infected < 1:
-                    warnings.warn('Graph with less than 100 nodes: a single node will be set as infected')
+                    warnings.warn(
+                        "The fraction_infected value is too low given the number of nodes of the selected graph: a "
+                        "single node will be set as infected")
                     number_of_initial_infected = 1
 
                 available_nodes = [n for n in self.status if self.status[n] == 0]