Skip to content

Commit 03966d3

Browse files
committed
use * instead of comprehension
1 parent 0ff4697 commit 03966d3

File tree

1 file changed

+14
-13
lines changed

1 file changed

+14
-13
lines changed

ciw/import_params.py

Lines changed: 14 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def create_network_from_dictionary(params_input):
7878
number_of_classes = params["number_of_classes"]
7979
number_of_nodes = params["number_of_nodes"]
8080
if isinstance(params["priority_classes"], dict):
81-
preempt_priorities = [False for _ in range(number_of_nodes)]
81+
preempt_priorities = [False] * number_of_nodes
8282
if isinstance(params["priority_classes"], tuple):
8383
preempt_priorities = params["priority_classes"][1]
8484
params["priority_classes"] = {
@@ -87,7 +87,7 @@ def create_network_from_dictionary(params_input):
8787
}
8888
class_change_matrices = params.get(
8989
"class_change_matrices",
90-
[None for nd in range(number_of_nodes)],
90+
[None] * number_of_nodes,
9191
)
9292
class_change_time_distributions = {clss2: {clss1: None for clss1 in params['customer_class_names']} for clss2 in params['customer_class_names']}
9393
if 'class_change_time_distributions' in params:
@@ -165,28 +165,29 @@ def fill_out_dictionary(params):
165165
class_names = sorted(params["arrival_distributions"].keys())
166166
params["customer_class_names"] = class_names
167167

168+
number_of_nodes = len(params["number_of_servers"])
168169
default_dict = {
169170
"name": "Simulation",
170171
"routing": {class_name: routing.TransitionMatrix(transition_matrix=[[0.0]]) for class_name in class_names},
171-
"number_of_nodes": len(params["number_of_servers"]),
172+
"number_of_nodes": number_of_nodes,
172173
"number_of_classes": len(class_names),
173-
"queue_capacities": [float("inf") for _ in range(len(params["number_of_servers"]))],
174+
"queue_capacities": [float("inf")] * number_of_nodes,
174175
"priority_classes": {class_name: 0 for class_name in class_names},
175-
"baulking_functions": {class_name: [None for _ in range(len(params["number_of_servers"]))]for class_name in class_names},
176+
"baulking_functions": {class_name: [None] * number_of_nodes for class_name in class_names},
176177
"batching_distributions": {class_name: [
177-
ciw.dists.Deterministic(1) for _ in range(len(params["number_of_servers"]))
178-
] for class_name in class_names},
179-
"ps_thresholds": [1 for _ in range(len(params["number_of_servers"]))],
178+
ciw.dists.Deterministic(1)
179+
] * number_of_nodes for class_name in class_names},
180+
"ps_thresholds": [1] * number_of_nodes,
180181
"server_priority_functions": [
181-
None for _ in range(len(params["number_of_servers"]))
182-
],
182+
None
183+
] * number_of_nodes,
183184
"reneging_time_distributions": {
184-
class_name: [None for _ in range(len(params["number_of_servers"]))]
185+
class_name: [None] * number_of_nodes
185186
for class_name in class_names
186187
},
187188
"service_disciplines": [
188-
ciw.disciplines.FIFO for _ in range(len(params["number_of_servers"]))
189-
],
189+
ciw.disciplines.FIFO
190+
] * number_of_nodes,
190191
"system_capacity": float('inf')
191192
}
192193

0 commit comments

Comments
 (0)