Skip to content
4 changes: 2 additions & 2 deletions src/snakemake_interface_scheduler_plugins/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
__license__ = "MIT"

import logging
from typing import Dict, Mapping, Optional, Union, Sequence
from typing import Dict, Mapping, Optional, Union, Sequence, Set
from snakemake_interface_scheduler_plugins.interfaces.dag import DAGSchedulerInterface
from snakemake_interface_scheduler_plugins.interfaces.jobs import JobSchedulerInterface
from snakemake_interface_scheduler_plugins.settings import (
Expand Down Expand Up @@ -44,7 +44,7 @@ def select_jobs(
remaining_jobs: Sequence[JobSchedulerInterface],
available_resources: Mapping[str, Union[int, str]],
input_sizes: Dict[AnnotatedStringInterface, int],
) -> Optional[Sequence[JobSchedulerInterface]]:
) -> Optional[Set[JobSchedulerInterface]]:
"""Select jobs from the selectable jobs sequence. Thereby, ensure that the selected
jobs do not exceed the available resources.

Expand Down
6 changes: 3 additions & 3 deletions src/snakemake_interface_scheduler_plugins/tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -128,7 +128,7 @@ def test_scheduler(self):
available_resources={"cpu": 1, "mem_mb": 1024},
input_sizes=defaultdict(int),
)
assert scheduled == [], (
assert scheduled == set([]), (
"Scheduler should not select jobs exceeding available resources"
)

Expand All @@ -138,7 +138,7 @@ def test_scheduler(self):
available_resources={"cpu": 1, "mem_mb": 2048},
input_sizes=defaultdict(int),
)
assert scheduled == [dag._jobs[0]], "Scheduler did not select the expected job"
assert scheduled == set([dag._jobs[0]]), "Scheduler did not select the expected job"

dag._finished.add(dag._jobs[0])

Expand All @@ -148,6 +148,6 @@ def test_scheduler(self):
available_resources={"cpu": 5, "mem_mb": 10000},
input_sizes=defaultdict(int),
)
assert scheduled == [dag._jobs[1], dag._jobs[2]], (
assert scheduled == set([dag._jobs[1], dag._jobs[2]]), (
"Scheduler did not select the expected jobs"
)
Loading