enable random

model/model_manager.py

@@ -70,12 +70,17 @@ class ModelManager:
 
     def generate_image_ig_parallel_anony(self, prompt, model_A, model_B, model_C, model_D):
         if model_A == "" and model_B == "" and model_C == "" and model_D == "":
-            # model_names = random.sample([model for model in self.model_ig_list], 4)
-            from .matchmaker import matchmaker
-            model_ids = matchmaker(num_players=len(self.model_ig_list))
-            print(model_ids)
-            model_names = [self.model_ig_list[i] for i in model_ids]
-            print(model_names)
+            not_run = [11, 12, 13, 14, 15, 16, 17, 18, 19]
+            filtered_models = [model for i, model in enumerate(self.model_ig_list) if i not in not_run]
+            model_names = random.sample([model for model in filtered_models], 4)
+            
+
+
+            # from .matchmaker import matchmaker
+            # model_ids = matchmaker(num_players=len(self.model_ig_list))
+            # print(model_ids)
+            # model_names = [self.model_ig_list[i] for i in model_ids]
+            # print(model_names)
         else:
             model_names = [model_A, model_B, model_C, model_D]
 
@@ -90,11 +95,16 @@ class ModelManager:
     def generate_image_ig_museum_parallel_anony(self, model_A, model_B, model_C, model_D):
         if model_A == "" and model_B == "" and model_C == "" and model_D == "":
             # model_names = random.sample([model for model in self.model_ig_list], 4)
-            from .matchmaker import matchmaker
-            model_ids = matchmaker(num_players=len(self.model_ig_list))
-            print(model_ids)
-            model_names = [self.model_ig_list[i] for i in model_ids]
-            print(model_names)
+
+            not_run = [11, 12, 13, 14, 15, 16, 17, 18, 19]
+            filtered_models = [model for i, model in enumerate(self.model_ig_list) if i not in not_run]
+            model_names = random.sample([model for model in filtered_models], 4)
+
+            # from .matchmaker import matchmaker
+            # model_ids = matchmaker(num_players=len(self.model_ig_list))
+            # print(model_ids)
+            # model_names = [self.model_ig_list[i] for i in model_ids]
+            # print(model_names)
         else:
             model_names = [model_A, model_B, model_C, model_D]
 

serve/update_skill.py

@@ -82,6 +82,7 @@ def update_skill(rank, model_names, k_group=4):
 
     ratings, comparison_counts, total_comparisons = load_json_via_sftp()
 
+    '''
     # group = Model_ID.group
     group = []
     for model_name in model_names:
@@ -101,7 +102,7 @@ def update_skill(rank, model_names, k_group=4):
             
         comparison_counts[group[player1], group[player2]] += 1
         comparison_counts[group[player2], group[player1]] += 1
-        
+    '''
     total_comparisons += 1
 
     save_json_via_sftp(ratings, comparison_counts, total_comparisons)