fix: type error

discopop-project · Jan 3, 2024 · ceb463a · ceb463a
1 parent 134a36d
commit ceb463a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/discopop_library/discopop_optimizer/optimization/evolutionary_algorithm.py b/discopop_library/discopop_optimizer/optimization/evolutionary_algorithm.py
@@ -458,7 +458,7 @@ def __dump_result(
 
 def __get_random_configuration(
     experiment: Experiment,
-    available_decisions: Dict[FunctionRoot, List[Set[int]]],
+    available_decisions: Dict[FunctionRoot, List[List[int]]],
     arguments: OptimizerArguments,
 ):
     while True:
@@ -468,7 +468,8 @@ def __get_random_configuration(
             excluded: Set[int] = set()
             requirements: Set[int] = set()
 
-            for decision_set in available_decisions[function]:
+            for decision_list in available_decisions[function]:
+                decision_set = set(decision_list)
                 decision_set = decision_set - (decision_set & excluded)
                 reduced_decision_set = decision_set.intersection(requirements)
                 if len(reduced_decision_set) != 0: