Remove: Functional graph workflow

This reverts commit f895c6b.
wysockipiotr · May 17, 2024 · d17dbfd · d17dbfd
1 parent f895c6b
commit d17dbfd
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 31 deletions.
diff --git a/smartschedule/planning/parallelization/stage_parallelization.py b/smartschedule/planning/parallelization/stage_parallelization.py
@@ -1,12 +1,15 @@
+from smartschedule.planning.parallelization.parallel_stages_list import (
+    ParallelStagesList,
+)
 from smartschedule.planning.parallelization.sorted_nodes_to_parallelized_stages import (
     sorted_nodes_to_parallelized_stages,
 )
+from smartschedule.planning.parallelization.stage import Stage
 from smartschedule.planning.parallelization.stages_to_nodes import stages_to_nodes
 from smartschedule.sorter import graph_topological_sort
-from smartschedule.utils.pipe import pipe
 
-stage_parallelization_of = pipe(
-    stages_to_nodes,
-    graph_topological_sort,
-    sorted_nodes_to_parallelized_stages,
-)
+
+def stage_parallelization_of(stages: set[Stage]) -> ParallelStagesList:
+    nodes = stages_to_nodes(stages)
+    sorted_nodes = graph_topological_sort(nodes)
+    return sorted_nodes_to_parallelized_stages(sorted_nodes)
diff --git a/smartschedule/utils/__init__.py b/smartschedule/utils/__init__.py
diff --git a/smartschedule/utils/pipe.py b/smartschedule/utils/pipe.py