Update storage_classes support for arvados_cwl_runner to work correctly

[arvados.git] / sdk / cwl / arvados_cwl / arvcontainer.py

diff --git a/sdk/cwl/arvados_cwl/arvcontainer.py b/sdk/cwl/arvados_cwl/arvcontainer.py
index 0bec692643ad805c02d6b8358fae8a65841c1367..05c0212fc6eeae41b2a10279f04ebd71000f1eaf 100644 (file)
--- a/sdk/cwl/arvados_cwl/arvcontainer.py
+++ b/sdk/cwl/arvados_cwl/arvcontainer.py
@@ -427,6 +427,9 @@ class RunnerContainer(Runner):
         if kwargs.get("debug"):
             command.append("--debug")
 
+        if kwargs.get("storage_classes") and kwargs.get("storage_classes") != self.default_storage_classes:
+            command.append("--storage-classes=" + kwargs.get("storage_classes"))
+
         if self.on_error:
             command.append("--on-error=" + self.on_error)