Collect assets and flows in the model to make it faster to evaluate p…

…resence in the set
TulipaEnergy · Jan 17, 2024 · c0b12dd · c0b12dd
1 parent 5deb5a3
commit c0b12dd
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/create-model.jl b/src/create-model.jl
@@ -56,10 +56,10 @@ function create_model(graph, representative_periods, constraints_partitions; wri
     end
 
     ## Sets unpacking
-    A = labels(graph)
-    F = edge_labels(graph)
-    filter_assets(key, value) = Iterators.filter(a -> getfield(graph[a], key) == value, A)
-    filter_flows(key, value) = Iterators.filter(f -> getfield(graph[f...], key) == value, F)
+    A = labels(graph) |> collect
+    F = edge_labels(graph) |> collect
+    filter_assets(key, value) = filter(a -> getfield(graph[a], key) == value, A)
+    filter_flows(key, value) = filter(f -> getfield(graph[f...], key) == value, F)
 
     Ac = filter_assets(:type, "consumer")
     Ap = filter_assets(:type, "producer")