From 640fd49010c5967d90af64f059ef5b534d91cfff Mon Sep 17 00:00:00 2001
From: Petar Petrov <petarpetrov2020@outlook.com>
Date: Sun, 24 Nov 2024 16:56:07 +0100
Subject: [PATCH] 777

---
 src/mqt/qao/karp/karp_number.py | 30 ++++++++++++++++++++----------
 1 file changed, 20 insertions(+), 10 deletions(-)

diff --git a/src/mqt/qao/karp/karp_number.py b/src/mqt/qao/karp/karp_number.py
index 631487e..ce8883a 100644
--- a/src/mqt/qao/karp/karp_number.py
+++ b/src/mqt/qao/karp/karp_number.py
@@ -153,7 +153,11 @@ def sat(
 
         txt_outputname = "SAT: "
 
-        solution = solver_method(problem)
+
+        if isinstance(problem, Problem):
+            solution = solver_method(problem)
+        else:
+            raise TypeError("Expected `problem` to be of type `Problem`.")
 
         if solution is None or not hasattr(solution, "best_solution"):
             msg = "Solver did not return a valid solution."
@@ -164,7 +168,7 @@ def sat(
             k.split("_")[-1].replace("!", ""): (1.0 - v if "!" in k else v) for k, v in set_variables.items()
         }
 
-        value_dict = {}
+        value_dict: dict[Any, Any] = {}
         for k, v in output_dict.items():
             value_dict.setdefault(v, []).append(k)
 
@@ -204,7 +208,7 @@ def three_sat(
         solver_method: Callable[..., Any] | None = None,
         read_solution: Literal["print", "file"] | None = None,
         solver_params: dict[str, Any] | None = None,
-    ) -> Problem | dict[str, float]:
+    ) -> Problem | dict[str, float] | Union[Problem, list[int]]:
         """Initializes and optionally solves the 3-SAT (Satisfiability) problem.
 
         Args:
@@ -238,10 +242,13 @@ def three_sat(
                     literals = processed_line.split()
                     clauses.append(literals)
 
-        else:
-            clauses = input_data
+        elif all(isinstance(item, tuple) for item in input_data):
+            # Convert list[tuple[int, int]] to list[list[str]]
+            clauses = [[str(item[0]), str(item[1])] for item in input_data]
             filename = ""
-
+        else:
+            raise ValueError("Invalid input_data type. Expected str or list[tuple[int, int]].")
+        
         graph = KarpNumber._create_graph(clauses)
 
         problem = KarpGraphs.independent_set(graph, solve=False)
@@ -257,7 +264,10 @@ def three_sat(
 
         txt_outputname = "3-SAT: "
 
-        solution = solver_method(problem)
+        if isinstance(problem, Problem):
+            solution = solver_method(problem)
+        else:
+            raise TypeError("Expected `problem` to be of type `Problem`.")
 
         if solution is None or not hasattr(solution, "best_solution"):
             msg = "Solver did not return a valid solution."
@@ -268,7 +278,7 @@ def three_sat(
             k.split("_")[-1].replace("!", ""): (1.0 - v if "!" in k else v) for k, v in set_variables.items()
         }
 
-        value_dict = {}
+        value_dict: dict[Any, Any] = {}
         for k, v in output_dict.items():
             value_dict.setdefault(v, []).append(k)
 
@@ -303,8 +313,8 @@ def three_sat(
 
     @staticmethod
     def check_three_sat_solution(
-        clauses: list[list[str]], solution: dict[str, float]
-    ) -> dict[str, Union[bool, list[list[str]], dict[str, str]]]:
+        clauses: Any, solution: dict[str, float]
+    ) -> dict[Any, Any]:
         """Validates a solution for the 3-SAT problem by checking clause satisfaction."""
         not_satisfied_clauses = []