Add comments to structurator (#80)

aimclub · Feb 7, 2024 · 40e176e · 40e176e
1 parent 5775ac9
commit 40e176e
Show file tree

Hide file tree

Showing 3 changed files with 192 additions and 45 deletions.
diff --git a/sampo/scheduler/genetic/converter.py b/sampo/scheduler/genetic/converter.py
@@ -177,7 +177,7 @@ def decode(work_index):
     # declare current checkpoint index
     ckpt_idx = 0
     start_time = assigned_parent_time - 1
-    pred_start_time = start_time - 1
+    prev_start_time = start_time - 1
 
     def work_scheduled(args) -> bool:
         idx, (work_idx, node, worker_team, contractor, exec_time, work_spec) = args
@@ -209,13 +209,13 @@ def work_scheduled(args) -> bool:
     while len(enumerated_works_remaining) > 0:
         if ckpt_idx < len(work_timeline):
             start_time = work_timeline[ckpt_idx]
-            if pred_start_time == start_time:
+            if prev_start_time == start_time:
                 ckpt_idx += 1
                 continue
             if start_time.is_inf():
                 # break because schedule already contains Time.inf(), that is incorrect schedule
                 break
-            pred_start_time = start_time
+            prev_start_time = start_time
         else:
             start_time += 1
 

diff --git a/sampo/structurator/__init__.py b/sampo/structurator/__init__.py
@@ -1,4 +1,3 @@
 from sampo.structurator.base import graph_restructuring, STAGE_SEP
 from sampo.structurator.graph_insertion import graph_in_graph_insertion
 from sampo.structurator.light_modifications import work_graph_ids_simplification
-from sampo.structurator.base import graph_restructuring