diff --git a/tupak/__init__.py b/tupak/__init__.py index 285c2c448bc9fb13387568a6d68915e664e8dcb4..251b7fca71e80a40529380087bce773768781f49 100644 --- a/tupak/__init__.py +++ b/tupak/__init__.py @@ -1,10 +1,11 @@ from __future__ import print_function, division -# import local files +# import local files, utils should be imported first from . import utils from . import detector from . import prior from . import source from . import likelihood from . import waveform_generator -from . import sampler \ No newline at end of file +from . import sampler + diff --git a/tupak/utils.py b/tupak/utils.py index 866d8e89d5b9f62ef4adb88892202598a565d8a9..cd15302089101f7f509da80b448ab896b1c4ee04 100644 --- a/tupak/utils.py +++ b/tupak/utils.py @@ -543,6 +543,14 @@ def set_up_command_line_arguments(): command_line_args = set_up_command_line_arguments() +if 'DISPLAY' in os.environ: + pass +else: + logging.info('No $DISPLAY environment variable found, so importing \ + matplotlib.pyplot with non-interactive "Agg" backend.') + import matplotlib + matplotlib.use('Agg') +