Fix multithreading (#2198)

### Changes The `_multithreading_calculation_ranking_score` does not work when len(groups_to_rank) < self._num_workers ### Reason for changes IndexError: pop from empty list ### Related tickets N/A ### Tests N/A --------- Co-authored-by: Alexander Suslov <[email protected]>
openvinotoolkit · Oct 16, 2023 · 9b75974 · 9b75974
1 parent 4d47869
commit 9b75974
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nncf/quantization/algorithms/accuracy_control/ranker.py b/nncf/quantization/algorithms/accuracy_control/ranker.py
@@ -223,8 +223,8 @@ def _multithreading_calculation_ranking_score(
                 )
                 ranking_scores.append(float(ranking_score))
 
-        for _ in range(self._num_workers - 1):
-            prepared_model = prepared_model_queue.pop(0).result()
+        for future in prepared_model_queue:
+            prepared_model = future.result()
             ranking_score = self._calculate_ranking_score(
                 prepared_model, ranking_subset_indices, reference_values_for_each_item
             )