ML4CO-Kit 0.1.1

Thinklab-SJTU · Sep 3, 2024 · b89a387 · b89a387
1 parent 703ddc2
commit b89a387
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/ml4co_kit/__init__.py b/ml4co_kit/__init__.py
@@ -37,5 +37,5 @@
     from .learning.utils import points_to_distmat, sparse_points
 
 
-__version__ = "0.1.0"
+__version__ = "0.1.1"
 __author__ = "SJTU-ReThinkLab"
diff --git a/tests/test_algorithm.py b/tests/test_algorithm.py
@@ -19,7 +19,7 @@ def test_tsp_greedy_decoder():
     solver.from_txt("tests/algorithm_test/tsp50.txt")
     points = solver.points
     heatmap = np.load("tests/algorithm_test/tsp50_heatmap.npy", allow_pickle=True)
-    tours = tsp_greedy_decoder(heatmap=heatmap, points=points)
+    tours = tsp_greedy_decoder(heatmap=heatmap)
     solver.read_tours(tours)
     _, _, gap_avg, _ = solver.evaluate(calculate_gap=True)
     print(f"Greedy Decoder Gap: {gap_avg}")
@@ -63,7 +63,7 @@ def test_tsp_mcts_local_search():
     solver.from_txt("tests/algorithm_test/tsp50.txt")
     points = solver.points
     heatmap = np.load("tests/algorithm_test/tsp50_heatmap.npy", allow_pickle=True)
-    greedy_tours = tsp_greedy_decoder(heatmap=heatmap, points=points)
+    greedy_tours = tsp_greedy_decoder(heatmap=heatmap)
     tours = tsp_mcts_local_search(
         init_tours=greedy_tours, heatmap=heatmap, points=points, time_limit=0.1
     )