job-shop-lib 1.0.0a5__py3-none-any.whl → 1.0.0b1__py3-none-any.whl

job_shop_lib/reinforcement_learning/_multi_job_shop_graph_env.py

@@ -2,7 +2,7 @@
 
 from collections import defaultdict
 from collections.abc import Callable, Sequence
-from typing import Any
+from typing import Any, Tuple, Dict, List, Optional, Type
 from copy import deepcopy
 
 import gymnasium as gym
@@ -16,7 +16,7 @@ from job_shop_lib.dispatching import (
 )
 from job_shop_lib.dispatching.feature_observers import FeatureObserverConfig
 from job_shop_lib.generation import InstanceGenerator
-from job_shop_lib.graphs import JobShopGraph, build_agent_task_graph
+from job_shop_lib.graphs import JobShopGraph, build_resource_task_graph
 from job_shop_lib.graphs.graph_updaters import (
     GraphUpdater,
     ResidualGraphUpdater,
@@ -160,18 +160,18 @@ class MultiJobShopGraphEnv(gym.Env):
         feature_observer_configs: Sequence[FeatureObserverConfig],
         graph_initializer: Callable[
             [JobShopInstance], JobShopGraph
-        ] = build_agent_task_graph,
+        ] = build_resource_task_graph,
         graph_updater_config: DispatcherObserverConfig[
-            type[GraphUpdater]
+            Type[GraphUpdater]
         ] = DispatcherObserverConfig(class_type=ResidualGraphUpdater),
         ready_operations_filter: Callable[
-            [Dispatcher, list[Operation]], list[Operation]
+            [Dispatcher, List[Operation]], List[Operation]
         ] = filter_dominated_operations,
         reward_function_config: DispatcherObserverConfig[
-            type[RewardObserver]
+            Type[RewardObserver]
         ] = DispatcherObserverConfig(class_type=MakespanReward),
-        render_mode: str | None = None,
-        render_config: RenderConfig | None = None,
+        render_mode: Optional[str] = None,
+        render_config: Optional[RenderConfig] = None,
         use_padding: bool = True,
     ) -> None:
         super().__init__()
@@ -226,7 +226,7 @@ class MultiJobShopGraphEnv(gym.Env):
     @property
     def ready_operations_filter(
         self,
-    ) -> Callable[[Dispatcher, list[Operation]], list[Operation]] | None:
+    ) -> Optional[Callable[[Dispatcher, List[Operation]], List[Operation]]]:
         """Returns the current ready operations filter."""
         return (
             self.single_job_shop_graph_env.dispatcher.ready_operations_filter
@@ -236,7 +236,7 @@ class MultiJobShopGraphEnv(gym.Env):
     def ready_operations_filter(
         self,
         pruning_function: Callable[
-            [Dispatcher, list[Operation]], list[Operation]
+            [Dispatcher, List[Operation]], List[Operation]
         ],
     ) -> None:
         """Sets the ready operations filter."""
@@ -267,9 +267,9 @@ class MultiJobShopGraphEnv(gym.Env):
     def reset(
         self,
         *,
-        seed: int | None = None,
-        options: dict[str, Any] | None = None,
-    ) -> tuple[ObservationDict, dict]:
+        seed: Optional[int] = None,
+        options: Dict[str, Any] | None = None,
+    ) -> Tuple[ObservationDict, Dict[str, Any]]:
         """Resets the environment and returns the initial observation.
 
         Args:
@@ -303,8 +303,8 @@ class MultiJobShopGraphEnv(gym.Env):
         return obs, info
 
     def step(
-        self, action: tuple[int, int]
-    ) -> tuple[ObservationDict, float, bool, bool, dict]:
+        self, action: Tuple[int, int]
+    ) -> Tuple[ObservationDict, float, bool, bool, Dict[str, Any]]:
         """Takes a step in the environment.
 
         Args:
@@ -322,9 +322,10 @@ class MultiJobShopGraphEnv(gym.Env):
             - Whether the environment is done.
             - Whether the episode was truncated (always False).
             - A dictionary with additional information. The dictionary
-              contains the following keys: ``"feature_names"``, The names of
-              the features in the observation; ``"available_operations"``, the
-              operations that are ready to be scheduled.
+              contains the following keys: "feature_names", the names of the
+              features in the observation; and "available_operations_with_ids",
+              a list of available actions in the form of (operation_id,
+              machine_id, job_id).
         """
         obs, reward, done, truncated, info = (
             self.single_job_shop_graph_env.step(action)
@@ -355,7 +356,7 @@ class MultiJobShopGraphEnv(gym.Env):
             input_shape: (num_machines, num_features)
             output_shape: (max_num_machines, num_features) (padded with -1)
         """
-        padding_value: dict[str, float | bool] = defaultdict(lambda: -1)
+        padding_value: Dict[str, float | bool] = defaultdict(lambda: -1)
         padding_value[ObservationSpaceKey.REMOVED_NODES.value] = True
         for key, value in observation.items():
             if not isinstance(value, np.ndarray):  # Make mypy happy
@@ -368,7 +369,7 @@ class MultiJobShopGraphEnv(gym.Env):
             )
         return observation
 
-    def _get_output_shape(self, key: str) -> tuple[int, ...]:
+    def _get_output_shape(self, key: str) -> Tuple[int, ...]:
         """Returns the output shape of the observation space key."""
         output_shape = self.observation_space[key].shape
         assert output_shape is not None  # Make mypy happy
@@ -376,3 +377,22 @@ class MultiJobShopGraphEnv(gym.Env):
 
     def render(self) -> None:
         self.single_job_shop_graph_env.render()
+
+    def get_available_actions_with_ids(self) -> List[Tuple[int, int, int]]:
+        """Returns a list of available actions in the form of
+        (operation_id, machine_id, job_id)."""
+        return self.single_job_shop_graph_env.get_available_actions_with_ids()
+
+    def validate_action(self, action: Tuple[int, int]) -> None:
+        """Validates the action.
+
+        Args:
+            action:
+                The action to validate. The action is a tuple of two integers
+                (job_id, machine_id): the job ID and the machine ID in which
+                to schedule the operation.
+
+        Raises:
+            ValidationError: If the action is invalid.
+        """
+        self.single_job_shop_graph_env.validate_action(action)

job_shop_lib/reinforcement_learning/_reward_observers.py

@@ -1,6 +1,8 @@
 """Rewards functions are defined as `DispatcherObervers` and are used to
 calculate the reward for a given state."""
 
+from typing import List
+
 from job_shop_lib.dispatching import DispatcherObserver, Dispatcher
 from job_shop_lib import ScheduledOperation
 
@@ -18,7 +20,7 @@ class RewardObserver(DispatcherObserver):
         self, dispatcher: Dispatcher, *, subscribe: bool = True
     ) -> None:
         super().__init__(dispatcher, subscribe=subscribe)
-        self.rewards: list[float] = []
+        self.rewards: List[float] = []
 
     @property
     def last_reward(self) -> float:

job_shop_lib/reinforcement_learning/_single_job_shop_graph_env.py

@@ -2,7 +2,7 @@
 
 from copy import deepcopy
 from collections.abc import Callable, Sequence
-from typing import Any
+from typing import Any, Dict, Tuple, List, Optional, Type
 
 import matplotlib.pyplot as plt
 import gymnasium as gym
@@ -15,6 +15,7 @@ from job_shop_lib.graphs.graph_updaters import (
     GraphUpdater,
     ResidualGraphUpdater,
 )
+from job_shop_lib.exceptions import ValidationError
 from job_shop_lib.dispatching import (
     Dispatcher,
     filter_dominated_operations,
@@ -24,7 +25,7 @@ from job_shop_lib.dispatching.feature_observers import (
     FeatureObserverConfig,
     CompositeFeatureObserver,
 )
-from job_shop_lib.visualization import GanttChartCreator
+from job_shop_lib.visualization.gantt import GanttChartCreator
 from job_shop_lib.reinforcement_learning import (
     RewardObserver,
     MakespanReward,
@@ -138,16 +139,16 @@ class SingleJobShopGraphEnv(gym.Env):
         job_shop_graph: JobShopGraph,
         feature_observer_configs: Sequence[FeatureObserverConfig],
         reward_function_config: DispatcherObserverConfig[
-            type[RewardObserver]
+            Type[RewardObserver]
         ] = DispatcherObserverConfig(class_type=MakespanReward),
         graph_updater_config: DispatcherObserverConfig[
-            type[GraphUpdater]
+            Type[GraphUpdater]
         ] = DispatcherObserverConfig(class_type=ResidualGraphUpdater),
-        ready_operations_filter: (
-            Callable[[Dispatcher, list[Operation]], list[Operation]] | None
-        ) = filter_dominated_operations,
-        render_mode: str | None = None,
-        render_config: RenderConfig | None = None,
+        ready_operations_filter: Optional[
+            Callable[[Dispatcher, List[Operation]], List[Operation]]
+        ] = filter_dominated_operations,
+        render_mode: Optional[str] = None,
+        render_config: Optional[RenderConfig] = None,
         use_padding: bool = True,
     ) -> None:
         super().__init__()
@@ -195,10 +196,26 @@ class SingleJobShopGraphEnv(gym.Env):
         """Returns the job shop graph."""
         return self.graph_updater.job_shop_graph
 
+    def current_makespan(self) -> int:
+        """Returns current makespan of partial schedule."""
+        return self.dispatcher.schedule.makespan()
+
+    def machine_utilization(self) -> NDArray[np.float32]:
+        """Returns utilization percentage for each machine."""
+        total_time = max(1, self.current_makespan())  # Avoid division by zero
+        machine_busy_time = np.zeros(self.instance.num_machines)
+
+        for m_id, m_schedule in enumerate(self.dispatcher.schedule.schedule):
+            machine_busy_time[m_id] = sum(
+                op.operation.duration for op in m_schedule
+            )
+
+        return machine_busy_time / total_time
+
     def _get_observation_space(self) -> gym.spaces.Dict:
         """Returns the observation space dictionary."""
         num_edges = self.job_shop_graph.num_edges
-        dict_space: dict[str, gym.Space] = {
+        dict_space: Dict[str, gym.Space] = {
             ObservationSpaceKey.REMOVED_NODES.value: gym.spaces.MultiBinary(
                 len(self.job_shop_graph.nodes)
             ),
@@ -224,18 +241,23 @@ class SingleJobShopGraphEnv(gym.Env):
     def reset(
         self,
         *,
-        seed: int | None = None,
-        options: dict[str, Any] | None = None,
-    ) -> tuple[ObservationDict, dict]:
+        seed: Optional[int] = None,
+        options: Optional[Dict[str, Any]] = None,
+    ) -> Tuple[ObservationDict, dict]:
         """Resets the environment."""
         super().reset(seed=seed, options=options)
         self.dispatcher.reset()
         obs = self.get_observation()
-        return obs, {}
+        return obs, {
+            "feature_names": self.composite_observer.column_names,
+            "available_operations_with_ids": (
+                self.get_available_actions_with_ids()
+            ),
+        }
 
     def step(
-        self, action: tuple[int, int]
-    ) -> tuple[ObservationDict, float, bool, bool, dict[str, Any]]:
+        self, action: Tuple[int, int]
+    ) -> Tuple[ObservationDict, float, bool, bool, Dict[str, Any]]:
         """Takes a step in the environment.
 
         Args:
@@ -254,9 +276,9 @@ class SingleJobShopGraphEnv(gym.Env):
             - Whether the episode was truncated (always False).
             - A dictionary with additional information. The dictionary
               contains the following keys: "feature_names", the names of the
-              features in the observation; "available_operations", the
-              operations that are ready to be scheduled.
-
+              features in the observation; and "available_operations_with_ids",
+              a list of available actions in the form of (operation_id,
+              machine_id, job_id).
         """
         job_id, machine_id = action
         operation = self.dispatcher.next_operation(job_id)
@@ -269,9 +291,11 @@ class SingleJobShopGraphEnv(gym.Env):
         reward = self.reward_function.last_reward
         done = self.dispatcher.schedule.is_complete()
         truncated = False
-        info: dict[str, Any] = {
+        info: Dict[str, Any] = {
             "feature_names": self.composite_observer.column_names,
-            "available_operations": self.dispatcher.available_operations(),
+            "available_operations_with_ids": (
+                self.get_available_actions_with_ids()
+            ),
         }
         return obs, reward, done, truncated, info
 
@@ -322,6 +346,49 @@ class SingleJobShopGraphEnv(gym.Env):
         elif self.render_mode == "save_gif":
             self.gantt_chart_creator.create_gif()
 
+    def get_available_actions_with_ids(self) -> List[Tuple[int, int, int]]:
+        """Returns a list of available actions in the form of
+        (operation_id, machine_id, job_id)."""
+        available_operations = self.dispatcher.available_operations()
+        available_operations_with_ids = []
+        for operation in available_operations:
+            job_id = operation.job_id
+            operation_id = operation.operation_id
+            for machine_id in operation.machines:
+                available_operations_with_ids.append(
+                    (operation_id, machine_id, job_id)
+                )
+        return available_operations_with_ids
+
+    def validate_action(self, action: Tuple[int, int]) -> None:
+        """Validates that the action is legal in the current state.
+
+        Args:
+            action:
+                The action to validate. The action is a tuple of two integers
+                (job_id, machine_id).
+
+        Raises:
+            ValidationError: If the action is invalid.
+        """
+        job_id, machine_id = action
+        if not 0 <= job_id < self.instance.num_jobs:
+            raise ValidationError(f"Invalid job_id {job_id}")
+
+        if not -1 <= machine_id < self.instance.num_machines:
+            raise ValidationError(f"Invalid machine_id {machine_id}")
+
+        # Check if job has operations left
+        job = self.instance.jobs[job_id]
+        if self.dispatcher.job_next_operation_index[job_id] >= len(job):
+            raise ValidationError(f"Job {job_id} has no operations left")
+
+        next_operation = self.dispatcher.next_operation(job_id)
+        if machine_id == -1 and len(next_operation.machines) > 1:
+            raise ValidationError(
+                f"Operation {next_operation} requires a machine_id"
+            )
+
 
 if __name__ == "__main__":
     from job_shop_lib.dispatching.feature_observers import (
@@ -333,7 +400,7 @@ if __name__ == "__main__":
 
     instance = load_benchmark_instance("ft06")
     job_shop_graph_ = build_disjunctive_graph(instance)
-    feature_observer_configs_ = [
+    feature_observer_configs_: List[DispatcherObserverConfig] = [
         DispatcherObserverConfig(
             FeatureObserverType.IS_READY,
             kwargs={"feature_types": [FeatureType.JOBS]},

job_shop_lib/reinforcement_learning/_types_and_constants.py

@@ -7,7 +7,7 @@ from typing import TypedDict
 import numpy as np
 
 from job_shop_lib.dispatching.feature_observers import FeatureType
-from job_shop_lib.visualization import (
+from job_shop_lib.visualization.gantt import (
     PartialGanttChartPlotterConfig,
     GifConfig,
     VideoConfig,

job_shop_lib/reinforcement_learning/_utils.py

@@ -1,6 +1,6 @@
 """Utility functions for reinforcement learning."""
 
-from typing import TypeVar, Any
+from typing import TypeVar, Any, Tuple, Optional, Type
 
 import numpy as np
 from numpy.typing import NDArray
@@ -12,9 +12,9 @@ T = TypeVar("T", bound=np.number)
 
 def add_padding(
     array: NDArray[Any],
-    output_shape: tuple[int, ...],
+    output_shape: Tuple[int, ...],
     padding_value: float = -1,
-    dtype: type[T] | None = None,
+    dtype: Optional[Type[T]] = None,
 ) -> NDArray[T]:
     """Adds padding to the array.
 

job_shop_lib/visualization/__init__.py

@@ -1,60 +0,0 @@
-"""Contains functions and classes for visualizing job shop scheduling problems.
-
-.. autosummary::
-
-    plot_gantt_chart
-    get_partial_gantt_chart_plotter
-    PartialGanttChartPlotter
-    create_gantt_chart_video
-    create_gantt_chart_gif
-    plot_disjunctive_graph
-    plot_agent_task_graph
-    GanttChartCreator
-    GifConfig
-    VideoConfig
-
-"""
-
-from job_shop_lib.visualization._plot_gantt_chart import plot_gantt_chart
-from job_shop_lib.visualization._gantt_chart_video_and_gif_creation import (
-    create_gantt_chart_gif,
-    create_gantt_chart_video,
-    create_gantt_chart_frames,
-    get_partial_gantt_chart_plotter,
-    create_video_from_frames,
-    create_gif_from_frames,
-    PartialGanttChartPlotter,
-)
-from job_shop_lib.visualization._plot_disjunctive_graph import (
-    plot_disjunctive_graph,
-    duration_labeler,
-)
-from job_shop_lib.visualization._plot_agent_task_graph import (
-    plot_agent_task_graph,
-    three_columns_layout,
-)
-from job_shop_lib.visualization._gantt_chart_creator import (
-    GanttChartCreator,
-    PartialGanttChartPlotterConfig,
-    GifConfig,
-    VideoConfig,
-)
-
-__all__ = [
-    "plot_gantt_chart",
-    "create_gantt_chart_video",
-    "create_gantt_chart_gif",
-    "create_gantt_chart_frames",
-    "get_partial_gantt_chart_plotter",
-    "create_gif_from_frames",
-    "create_video_from_frames",
-    "plot_disjunctive_graph",
-    "plot_agent_task_graph",
-    "three_columns_layout",
-    "GanttChartCreator",
-    "PartialGanttChartPlotterConfig",
-    "GifConfig",
-    "VideoConfig",
-    "PartialGanttChartPlotter",
-    "duration_labeler",
-]

job_shop_lib/visualization/gantt/__init__.py

@@ -0,0 +1,48 @@
+"""Contains functions and classes for visualizing job shop scheduling problems.
+
+.. autosummary::
+
+    plot_gantt_chart
+    get_partial_gantt_chart_plotter
+    PartialGanttChartPlotter
+    create_gantt_chart_video
+    create_gantt_chart_gif
+    GanttChartCreator
+    GifConfig
+    VideoConfig
+    PartialGanttChartPlotterConfig
+
+"""
+
+from ._plot_gantt_chart import plot_gantt_chart
+from ._gantt_chart_video_and_gif_creation import (
+    create_gantt_chart_gif,
+    create_gantt_chart_video,
+    create_gantt_chart_frames,
+    get_partial_gantt_chart_plotter,
+    create_video_from_frames,
+    create_gif_from_frames,
+    PartialGanttChartPlotter,
+)
+
+from ._gantt_chart_creator import (
+    GanttChartCreator,
+    PartialGanttChartPlotterConfig,
+    GifConfig,
+    VideoConfig,
+)
+
+__all__ = [
+    "plot_gantt_chart",
+    "create_gantt_chart_video",
+    "create_gantt_chart_gif",
+    "create_gantt_chart_frames",
+    "get_partial_gantt_chart_plotter",
+    "create_gif_from_frames",
+    "create_video_from_frames",
+    "GanttChartCreator",
+    "PartialGanttChartPlotterConfig",
+    "GifConfig",
+    "VideoConfig",
+    "PartialGanttChartPlotter",
+]

job_shop_lib/visualization/{_gantt_chart_creator.py → gantt/_gantt_chart_creator.py}

@@ -1,13 +1,13 @@
 """Home of the `GanttChartCreator` class and its configuration types."""
 
-from typing import TypedDict
+from typing import TypedDict, Optional
 import matplotlib.pyplot as plt
 
 from job_shop_lib.dispatching import (
     Dispatcher,
     HistoryObserver,
 )
-from job_shop_lib.visualization import (
+from job_shop_lib.visualization.gantt import (
     create_gantt_chart_video,
     get_partial_gantt_chart_plotter,
     create_gantt_chart_gif,
@@ -24,7 +24,7 @@ class PartialGanttChartPlotterConfig(TypedDict, total=False):
         - :func:`get_partial_gantt_chart_plotter`
     """
 
-    title: str | None
+    title: Optional[str]
     """The title of the Gantt chart."""
 
     cmap: str
@@ -43,7 +43,7 @@ class GifConfig(TypedDict, total=False):
         :func:`create_gantt_chart_gif`
     """
 
-    gif_path: str | None
+    gif_path: Optional[str]
     """The path to save the GIF. It must end with '.gif'."""
 
     fps: int
@@ -52,7 +52,7 @@ class GifConfig(TypedDict, total=False):
     remove_frames: bool
     """Whether to remove the frames after creating the GIF."""
 
-    frames_dir: str | None
+    frames_dir: Optional[str]
     """The directory to store the frames."""
 
     plot_current_time: bool
@@ -68,7 +68,7 @@ class VideoConfig(TypedDict, total=False):
         :func:`create_gantt_chart_video`
     """
 
-    video_path: str | None
+    video_path: Optional[str]
     """The path to save the video. It must end with a valid video extension
     (e.g., '.mp4')."""
 
@@ -78,7 +78,7 @@ class VideoConfig(TypedDict, total=False):
     remove_frames: bool
     """Whether to remove the frames after creating the video."""
 
-    frames_dir: str | None
+    frames_dir: Optional[str]
     """The directory to store the frames."""
 
     plot_current_time: bool
@@ -164,11 +164,11 @@ class GanttChartCreator:
     def __init__(
         self,
         dispatcher: Dispatcher,
-        partial_gantt_chart_plotter_config: (
-            PartialGanttChartPlotterConfig | None
-        ) = None,
-        gif_config: GifConfig | None = None,
-        video_config: VideoConfig | None = None,
+        partial_gantt_chart_plotter_config: Optional[
+            PartialGanttChartPlotterConfig
+        ] = None,
+        gif_config: Optional[GifConfig] = None,
+        video_config: Optional[VideoConfig] = None,
     ):
         if gif_config is None:
             gif_config = {}

job_shop_lib/visualization/{_gantt_chart_video_and_gif_creation.py → gantt/_gantt_chart_video_and_gif_creation.py}

@@ -3,7 +3,7 @@
 import os
 import pathlib
 import shutil
-from typing import Sequence, Protocol
+from typing import Sequence, Protocol, Optional, List
 
 import imageio
 import matplotlib.pyplot as plt
@@ -22,7 +22,7 @@ from job_shop_lib.dispatching import (
     HistoryObserver,
 )
 from job_shop_lib.dispatching.rules import DispatchingRuleSolver
-from job_shop_lib.visualization._plot_gantt_chart import plot_gantt_chart
+from job_shop_lib.visualization.gantt._plot_gantt_chart import plot_gantt_chart
 
 
 # This class serves as a more meaningful type hint than simply:
@@ -43,9 +43,9 @@ class PartialGanttChartPlotter(Protocol):
     def __call__(
         self,
         schedule: Schedule,
-        makespan: int | None = None,
-        available_operations: list[Operation] | None = None,
-        current_time: int | None = None,
+        makespan: Optional[int] = None,
+        available_operations: Optional[List[Operation]] = None,
+        current_time: Optional[int] = None,
     ) -> Figure:
         """Plots a Gantt chart for an unfinished schedule.
 
@@ -65,7 +65,7 @@ class PartialGanttChartPlotter(Protocol):
 
 
 def get_partial_gantt_chart_plotter(
-    title: str | None = None,
+    title: Optional[str] = None,
     cmap: str = "viridis",
     show_available_operations: bool = False,
 ) -> PartialGanttChartPlotter:
@@ -92,9 +92,9 @@ def get_partial_gantt_chart_plotter(
 
     def plot_function(
         schedule: Schedule,
-        makespan: int | None = None,
-        available_operations: list | None = None,
-        current_time: int | None = None,
+        makespan: Optional[int] = None,
+        available_operations: Optional[List[Operation]] = None,
+        current_time: Optional[int] = None,
     ) -> Figure:
         fig, ax = plot_gantt_chart(
             schedule, title=title, cmap_name=cmap, xlim=makespan
@@ -133,14 +133,14 @@ def get_partial_gantt_chart_plotter(
 # pylint: disable=too-many-arguments
 def create_gantt_chart_gif(
     instance: JobShopInstance,
-    gif_path: str | None = None,
-    solver: DispatchingRuleSolver | None = None,
-    plot_function: PartialGanttChartPlotter | None = None,
+    gif_path: Optional[str] = None,
+    solver: Optional[DispatchingRuleSolver] = None,
+    plot_function: Optional[PartialGanttChartPlotter] = None,
     fps: int = 1,
     remove_frames: bool = True,
-    frames_dir: str | None = None,
+    frames_dir: Optional[str] = None,
     plot_current_time: bool = True,
-    schedule_history: Sequence[ScheduledOperation] | None = None,
+    schedule_history: Optional[Sequence[ScheduledOperation]] = None,
 ) -> None:
     """Creates a GIF of the schedule being built.
 
@@ -202,14 +202,14 @@ def create_gantt_chart_gif(
 # pylint: disable=too-many-arguments
 def create_gantt_chart_video(
     instance: JobShopInstance,
-    video_path: str | None = None,
-    solver: DispatchingRuleSolver | None = None,
-    plot_function: PartialGanttChartPlotter | None = None,
+    video_path: Optional[str] = None,
+    solver: Optional[DispatchingRuleSolver] = None,
+    plot_function: Optional[PartialGanttChartPlotter] = None,
     fps: int = 1,
     remove_frames: bool = True,
-    frames_dir: str | None = None,
+    frames_dir: Optional[str] = None,
     plot_current_time: bool = True,
-    schedule_history: Sequence[ScheduledOperation] | None = None,
+    schedule_history: Optional[Sequence[ScheduledOperation]] = None,
 ) -> None:
     """Creates a video of the schedule being built.
 
@@ -268,10 +268,10 @@ def create_gantt_chart_video(
 def create_gantt_chart_frames(
     frames_dir: str,
     instance: JobShopInstance,
-    solver: DispatchingRuleSolver | None,
+    solver: Optional[DispatchingRuleSolver],
     plot_function: PartialGanttChartPlotter,
     plot_current_time: bool = True,
-    schedule_history: Sequence[ScheduledOperation] | None = None,
+    schedule_history: Optional[Sequence[ScheduledOperation]] = None,
 ) -> None:
     """Creates frames of the Gantt chart for the schedule being built.
 
@@ -411,7 +411,7 @@ def resize_image_to_macro_block(
     return image
 
 
-def _load_images(frames_dir: str) -> list:
+def _load_images(frames_dir: str) -> List:
     frames = [
         os.path.join(frames_dir, frame)
         for frame in sorted(os.listdir(frames_dir))