Updated default command

valohai · Apr 7, 2021 · 6b53ed9 · 6b53ed9
1 parent 5a057c4
commit 6b53ed9
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/valohai/internals/yaml.py b/valohai/internals/yaml.py
@@ -23,14 +23,12 @@ def generate_step(
     config_step = Step(
         name=step,
         image=image,
-        command="python %s {parameters}" % relative_source_path,
+        command=[
+            "pip install -r requirements.txt",
+            "python %s {parameters}" % relative_source_path
+        ],
     )
 
-    config_step.command = [
-        "pip install -r requirements.txt", 
-        "python %s {parameters}" % relative_source_path
-    ]
-
     for key, value in parameters.items():
         config_step.parameters[key] = Parameter(
             name=key,