str

Author: jakevc

snakemake_executor_plugin_aws_batch/batch_job_builder.py

@@ -77,17 +77,23 @@ def build_job_definition(self):
             "jobRoleArn": self.settings.job_role,
             "privileged": True,
             "resourceRequirements": [
-                {"type": BATCH_JOB_RESOURCE_REQUIREMENT_TYPE.VCPU.value, "value": vcpu},
+                {
+                    "type": BATCH_JOB_RESOURCE_REQUIREMENT_TYPE.VCPU.value,
+                    "value": vcpu_str,
+                },
                 {
                     "type": BATCH_JOB_RESOURCE_REQUIREMENT_TYPE.MEMORY.value,
-                    "value": mem,
+                    "value": mem_str,
                 },
             ],
         }
 
-        if int(gpu) > 0:
+        if gpu > 0:
             container_properties["resourceRequirements"].append(
-                {"type": BATCH_JOB_RESOURCE_REQUIREMENT_TYPE.GPU.value, "value": gpu}
+                {
+                    "type": BATCH_JOB_RESOURCE_REQUIREMENT_TYPE.GPU.value,
+                    "value": gpu_str,
+                }
             )
 
         timeout = {"attemptDurationSeconds": self.settings.task_timeout}