Update salesman_traveling.py

travelpy/salesman_traveling.py

@@ -1,10 +1,14 @@
+"""
+Module for solving the Traveling Salesman Problem (TSP) using brute force.
+"""
+
 from itertools import permutations
 
 def city_permutation(cities):
     """Function that generates all the permutations of the cities to be traveled"""
-    all_permutations = permutations(range(1, cities)) 
+    all_permutations = permutations(range(1, cities))
     for permutation in all_permutations:
-        yield (0,) + permutation + (0,) 
+        yield (0,) + permutation + (0,)
 
 def cost(matrix_distance, route):
     """Gives you the prices of all the routes that we have generated in city permutations."""
@@ -15,9 +19,8 @@ def cost(matrix_distance, route):
 
 def salesman_traveling(cities, matrix_distance):
     """Solves the Traveling Salesman Problem (TSP) using brute force"""
-    opt_c = float('inf')  # Inicializar con un valor alto para comparación
+    opt_c = float('inf')
     opt_x = []
-
     for route in city_permutation(cities):
         c = cost(matrix_distance, route)
         if c < opt_c: