Fix by Leon Lan for #6

d-krupke · Jun 13, 2024 · 8622c5d · 8622c5d
1 parent 46a80ab
commit 8622c5d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cpsat_log_parser/blocks/solver.py b/cpsat_log_parser/blocks/solver.py
@@ -53,6 +53,9 @@ def get_number_of_workers(self) -> int:
         for line in self.lines:
             if line.startswith("Setting number of workers to"):
                 return int(line.strip().split(" ")[-1])
+        # If `num_workers` is set, the number of workers is not shown in the log.
+        if "num_workers" in self.get_parameters():
+            return int(self.get_parameters()["num_workers"])
         # If `num_search_workers` is set, the number of workers is not shown in the log.
         if "num_search_workers" in self.get_parameters():
             return int(self.get_parameters()["num_search_workers"])