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