Skip to content

Commit 50e496f

Browse files
Fixed high memory requirements on carnival_brute_force.py
1 parent d3d4d09 commit 50e496f

1 file changed

Lines changed: 1 addition & 2 deletions

File tree

ch06-swarm_intelligence-ants/carnival_brute_force.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@
88
# Load the carnival attraction distances from a CSV file
99
attraction_count = 48
1010
attraction_data_file = 'attractions-' + str(attraction_count) + '.csv'
11-
attraction_permutations = set(itertools.permutations(range(0, attraction_count)))
1211
attraction_distances = []
1312
with open(attraction_data_file) as file:
1413
reader = csv.reader(file, quoting=csv.QUOTE_NONNUMERIC)
@@ -20,7 +19,7 @@
2019
best_distance = math.inf
2120
best_permutation = None
2221
# Determine the distance score for every permutation to find the best
23-
for attraction_permutation in attraction_permutations:
22+
for attraction_permutation in itertools.permutations(range(0, attraction_count)):
2423
last_attraction = attraction_permutation[0]
2524
total_distance = 0
2625
for attraction_index in range(1, len(attraction_permutation)):

0 commit comments

Comments
 (0)