diff options
Diffstat (limited to 'lib/spack/spack/cmd/modules/__init__.py')
-rw-r--r-- | lib/spack/spack/cmd/modules/__init__.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/lib/spack/spack/cmd/modules/__init__.py b/lib/spack/spack/cmd/modules/__init__.py index 00f008b2cf..f8f883e42e 100644 --- a/lib/spack/spack/cmd/modules/__init__.py +++ b/lib/spack/spack/cmd/modules/__init__.py @@ -8,6 +8,7 @@ import collections import os.path import shutil +import sys from llnl.util import filesystem, tty @@ -104,7 +105,7 @@ def one_spec_or_raise(specs): return specs[0] -def loads(module_type, specs, args): +def loads(module_type, specs, args, out=sys.stdout): """Prompt the list of modules associated with a list of specs""" # Get a comprehensive list of specs @@ -147,7 +148,8 @@ def loads(module_type, specs, args): d['comment'] = '' if not args.shell else '# {0}\n'.format( spec.format()) d['name'] = mod - print(prompt_template.format(**d)) + out.write(prompt_template.format(**d)) + out.write('\n') def find(module_type, specs, args): |