From 11af5c524e4eafdc3652921452663a50d9c36ec0 Mon Sep 17 00:00:00 2001
From: Teemu Ruokolainen <teemu.p.ruokolainen@gmail.com>
Date: Thu, 14 Mar 2024 13:53:21 +0200
Subject: [PATCH] fix

---
 src/submit.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/submit.py b/src/submit.py
index 1fe57a8..1fe75f9 100644
--- a/src/submit.py
+++ b/src/submit.py
@@ -262,7 +262,7 @@ def estimate_job_time(input_path: PosixPath) -> str:
 def create_sbatch_script_for_array_job(
     input_file: str,
     job_name: Path,
-    mem: int,
+    mem: str,
     cpus_per_task: int,
     time: str,
     email: str,
@@ -277,7 +277,7 @@ def create_sbatch_script_for_array_job(
         The json file with the list of audio files to process.
     job_name: Path
         The job name extracted from the input path.
-    mem: int
+    mem: str
         Requested memory per job. Default is 8GB.
     cpus_per_task: int
         Requested cpus per task. Default is 6.
@@ -298,7 +298,7 @@ def create_sbatch_script_for_array_job(
 #SBATCH --output="{tmp_dir}/{job_name}_%A_%a.out"
 #SBATCH --error="{tmp_dir}/{job_name}_%A_%a.err"
 #SBATCH --job-name={job_name}
-#SBATCH --mem={mem}G
+#SBATCH --mem={mem}
 #SBATCH --cpus-per-task={cpus_per_task}
 #SBATCH --gres=gpu:1
 #SBATCH --time={time}
@@ -369,7 +369,7 @@ def create_sbatch_script_for_single_file(
 #SBATCH --job-name={job_name}
 #SBATCH --output="{tmp_dir}/{job_name}_%j.out"
 #SBATCH --error="{tmp_dir}/{job_name}_%j.err"
-#SBATCH --mem={mem}G
+#SBATCH --mem={mem}
 #SBATCH --cpus-per-task={cpus_per_task}
 #SBATCH --gres=gpu:1
 #SBATCH --time={time}