Skip to content

Commit

Permalink
feature: add stream_type to RUN command like in EXEC
Browse files Browse the repository at this point in the history
  • Loading branch information
pustoshilov-d authored Aug 27, 2024
1 parent 2b22233 commit 625b5d9
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion spython/main/run.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@ def run(
return_result=False,
quiet=False,
background=False,
stream_type="stdout",
):
"""
run will run the container, with or withour arguments (which
Expand All @@ -49,6 +50,7 @@ def run(
return_result: if True, return entire json object with return code
and message result (default is False)
quiet: print the command to the user
stream_type: Sets which output stream from the singularity command should be return. Values are 'stdout', 'stderr', 'both'.
"""
from spython.utils import check_install

Expand Down Expand Up @@ -107,7 +109,7 @@ def run(
elif not stream:
result = self._run_command(cmd, sudo=sudo, return_result=return_result)
else:
return stream_command(cmd, sudo=sudo)
return stream_command(cmd, sudo=sudo, output_type=stream_type)

# If the user wants the raw result object
if return_result:
Expand Down

0 comments on commit 625b5d9

Please sign in to comment.