fix: restored previous function

Author: cmeesters

snakemake_executor_plugin_slurm/utils.py

@@ -1,5 +1,6 @@
 # utility functions for the SLURM executor plugin
 
+from collections import Counter
 import math
 import os
 import shlex
@@ -8,6 +9,7 @@
 from pathlib import Path
 from typing import Union
 
+from snakemake_interface_executor_plugins.dag import DAGExecutorInterface
 from snakemake_interface_executor_plugins.jobs import (
     JobExecutorInterface,
 )
@@ -69,6 +71,12 @@ def get_job_wildcards(job: JobExecutorInterface) -> str:
     return wildcard_str
 
 
+def pending_jobs_for_rule(dag: DAGExecutorInterface, rule_name: str) -> int:
+    """Count pending jobs for a given rule in the DAG."""
+    counts = Counter(job.rule.name for job in dag.needrun_jobs())
+    return counts.get(rule_name, 0)
+
+
 def round_half_up(n):
     return int(math.floor(n + 0.5))