⬆️ backward compatibility

ndlib/models/DiffusionModel.py

@@ -110,7 +110,8 @@ def __validate_configuration(self, configuration):
 
         # Checking initial simulation status
         sts = set(configuration.get_model_configuration().keys())
-        if self.discrete_state and "Infected" not in sts and "fraction_infected" not in mdp:
+        if self.discrete_state and "Infected" not in sts and "fraction_infected" not in mdp \
+                and "percentage_infected" not in mdp:
             warnings.warn('Initial infection missing: a random sample of 5% of graph nodes will be set as infected')
             self.params['model']["fraction_infected"] = 0.05
 
@@ -155,6 +156,8 @@ def set_initial_status(self, configuration):
 
         # Handle initial infection
         if 'Infected' not in self.params['status']:
+            if 'percentage_infected' in self.params['model']:
+                self.params['model']['fraction_infected'] = self.params['model']['percentage_infected']
             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:
@@ -219,6 +222,8 @@ def reset(self, infected_nodes=None):
             self.initial_status = self.status
 
         else:
+            if 'percentage_infected' in self.params['model']:
+                self.params['model']['fraction_infected'] = self.params['model']['percentage_infected']
             if 'fraction_infected' in self.params['model']:
                 for n in self.status:
                     self.status[n] = 0

ndlib/test/test_ndlib.py

@@ -123,7 +123,7 @@ def test_sir_model(self):
         config = mc.Configuration()
         config.add_model_parameter('beta', 0.5)
         config.add_model_parameter('gamma', 0.2)
-        config.add_model_parameter("fraction_infected", 0.1)
+        config.add_model_parameter("percentage_infected", 0.1)
         model.set_initial_status(config)
         iterations = model.iteration_bunch(10)
         self.assertEqual(len(iterations), 10)

setup.py

@@ -14,7 +14,7 @@
 
 
 setup(name='ndlib',
-      version='4.0.2',
+      version='5.0.0',
       license='BSD-Clause-2',
       description='Network Diffusion Library',
       url='https://github.com/GiulioRossetti/ndlib',