Some pep8 improvements

Author: geraintpalmer

ciw/arrival_node.py

@@ -55,7 +55,7 @@ def find_next_event_date(self):
         """
         Finds the time of the next arrival.
         """
-        times = [[self.event_dates_dict[nd+1][clss]
+        times = [[self.event_dates_dict[nd + 1][clss]
             for clss in range(len(self.event_dates_dict[1]))]
             for nd in range(len(self.event_dates_dict))]
 

ciw/node.py

@@ -201,7 +201,7 @@ def change_shift(self):
         Add servers and deletes or indicates which servers
         should go off duty.
         """
-        shift = self.next_event_date%self.cyclelength
+        shift = self.next_event_date % self.cyclelength
 
         try: indx = self.schedule.index(shift)
         except:

ciw/simulation.py

@@ -183,7 +183,7 @@ def find_times_dict(self, kind):
         Create the dictionary of service and arrival
         time functions for each node for each class
         """
-        return {node+1: {
+        return {node + 1: {
             clss: self.find_distributions(node, clss, kind)
             for clss in range(self.network.number_of_classes)}
             for node in range(self.network.number_of_nodes)}
@@ -193,7 +193,7 @@ def find_batches_dict(self):
         Create the dictionary of batch size
         functions for each node for each class
         """
-        return {node+1: {
+        return {node + 1: {
             clss: self.find_distributions(node, clss, 'Bch')
             for clss in range(self.network.number_of_classes)}
             for node in range(self.network.number_of_nodes)}
@@ -283,7 +283,7 @@ def simulate_until_max_time(self, max_simulation_time, progress_bar=False):
         next_active_node = self.find_next_active_node()
         current_time = next_active_node.next_event_date
 
-        if progress_bar is not False:
+        if progress_bar:
             self.progress_bar = tqdm.tqdm(total=max_simulation_time)
 
         while current_time < max_simulation_time:
@@ -318,7 +318,7 @@ def simulate_until_max_customers(self, max_customers,
         next_active_node = self.find_next_active_node()
         current_time = next_active_node.next_event_date
 
-        if progress_bar is not False:
+        if progress_bar:
             self.progress_bar = tqdm.tqdm(total=max_customers)
 
         if method == 'Finish':