Apply Gregs patch

bilby/__init__.py

@@ -24,4 +24,4 @@ from .core import utils, likelihood, prior, result, sampler
 from .core.sampler import run_sampler
 from .core.likelihood import Likelihood
 
-__version__ = '0.3.1'
+__version__ = utils.get_version_information()

bilby/core/utils.py

@@ -348,15 +348,21 @@ def setup_logger(outdir=None, label=None, log_level='INFO', print_version=False)
     for handler in logger.handlers:
         handler.setLevel(level)
 
-    version_file = os.path.join(
-        os.path.dirname(os.path.dirname(__file__)), '.version')
-    with open(version_file, 'r') as f:
-        version = f.readline().rstrip()
-
     if print_version:
+        version = get_version_information()
         logger.info('Running bilby version: {}'.format(version))
 
 
+def get_version_information():
+    version_file = os.path.join(
+        os.path.dirname(os.path.dirname(__file__)), '.version')
+    try:
+        with open(version_file, 'r') as f:
+            return f.readline().rstrip()
+    except: # noqa
+        return "No version information"
+
+
 def get_progress_bar(module='tqdm'):
     """
     TODO: Write proper docstring

setup.py

@@ -58,13 +58,7 @@ def readfile(filename):
     return filecontents
 
 
-VERSION_REGEX = re.compile(r"__version__ = \'(.*?)\'")
-CONTENTS = readfile(os.path.join(
-    os.path.dirname(os.path.abspath(__file__)),
-    "bilby", "__init__.py"))
-VERSION = VERSION_REGEX.findall(CONTENTS)[0]
-
-
+VERSION = '0.3.1'
 version_file = write_version_file(VERSION)
 long_description = get_long_description()