update submit.yml.erb (#25)

AaltoRSE · Jun 13, 2024 · dde6c39 · dde6c39
1 parent 484a71e
commit dde6c39
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/bin/deploy-data/ood/submit.yml.erb b/bin/deploy-data/ood/submit.yml.erb
@@ -10,5 +10,12 @@ batch_connect:
     %s
 
 script:
-  # Requesting 2Gb for each file
-  native: ["--mem", <%= (Dir.glob("#{audio_path}/*").select { |f| File.file?(f) }.count * 2).to_s + "G" %>]
+  # Request 2Gb per file
+  <% if File.directory?(audio_path) %>
+    native: ["--mem", <%= (Dir.glob("#{audio_path}/*").select { |f| File.file?(f) }.count * 2).to_s + "G" %>]
+  <% elsif File.file?(audio_path) %>
+    native: ["--mem", "2G"]
+  <% else %>
+    raise "Invalid audio_path: #{audio_path} is neither a file nor a directory."
+  <% end %>
+