fix

snakemake · Jul 20, 2024 · 6a8e08e · 6a8e08e
1 parent a0d2872
commit 6a8e08e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/snakemake_executor_plugin_azure_batch/__init__.py b/snakemake_executor_plugin_azure_batch/__init__.py
@@ -353,7 +353,7 @@ def run_job(self, job: JobExecutorInterface):
         self.logger.debug(f"Remote command: {remote_command}")
 
         task: bm.TaskAddParameter = build.batch_task(
-            job, self.envvars(), remote_command
+            job, self.container_image, self.envvars(), remote_command
         )
 
         job_info = SubmittedJobInfo(job, external_jobid=task.id)

diff --git a/snakemake_executor_plugin_azure_batch/build.py b/snakemake_executor_plugin_azure_batch/build.py
@@ -32,12 +32,8 @@
 
 from snakemake_executor_plugin_azure_batch.constant import DEFAULT_AUTO_SCALE_FORMULA
 
-from . import ExecutorSettings
 
-
-def batch_pool_params(
-    pool_id: str, settings: ExecutorSettings, container_image: str
-) -> Pool:
+def batch_pool_params(pool_id: str, settings, container_image: str) -> Pool:
     """
     Constructs a Batch Pool object with the specified parameters.