From 7d04822546d23e7e3ceddb32c6de07ca020bd9dc Mon Sep 17 00:00:00 2001
From: elad-c <elad.cohen@sony.com>
Date: Wed, 15 Jan 2025 18:44:54 +0200
Subject: [PATCH] fix

---
 .../core/common/graph/memory_graph/compute_graph_max_cut.py  | 5 -----
 1 file changed, 5 deletions(-)

diff --git a/model_compression_toolkit/core/common/graph/memory_graph/compute_graph_max_cut.py b/model_compression_toolkit/core/common/graph/memory_graph/compute_graph_max_cut.py
index 162c3b890..02a6c08ca 100644
--- a/model_compression_toolkit/core/common/graph/memory_graph/compute_graph_max_cut.py
+++ b/model_compression_toolkit/core/common/graph/memory_graph/compute_graph_max_cut.py
@@ -47,11 +47,6 @@ def compute_graph_max_cut(memory_graph: MemoryGraph,
     l_bound = memory_graph.memory_lbound_single_op
     u_bound = 2 * sum([t.total_size for t in memory_graph.b_nodes]) - l_bound
     it = 0
-
-    @timeout_decorator.timeout(300)
-    def solver_wrapper(*args, **kwargs):
-        return max_cut_astar.solve(*args, **kwargs)
-
     while it < n_iter:
         estimate = (u_bound + l_bound) / 2
         # Add a timeout of 5 minutes to the solver from the 2nd iteration.