improved function for extracting sub states

damianoazzolini · Nov 2, 2024 · e75aa2d · e75aa2d
1 parent 17f2055
commit e75aa2d
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/pastasolver/models_handler.py b/pastasolver/models_handler.py
@@ -675,11 +675,7 @@ def get_sub_world(super_w : str, map_id_list : 'list[int]') -> str:
         map_id_list = [0,2]
         result = 00 (extracts the values in position 0 and 2 of super_w)
         '''
-        sub_w : str = ""
-        for el in map_id_list:
-            sub_w = sub_w + super_w[el]
-
-        return sub_w
+        return ''.join([super_w[i] for i in map_id_list])
 
 
     def get_highest_prob_and_w_id_map(