fix pip install (#1686)

setup.cfg

@@ -74,6 +74,5 @@ exclude_lines =
 
 [options.entry_points]
 console_scripts =
-    tardis = tardis.scripts.tardis
     cmfgen2tardis = tardis.scripts.cmfgen2tardis:main
     tardis_test_runner = tardis.tests.integration_tests.runner:run_tests

setup.py

@@ -74,6 +74,10 @@
     version = '{version}'
 """.lstrip()
 
-setup(use_scm_version={'write_to': os.path.join('tardis', 'version.py'),
+# Pure Python scripts that are not entry points, required for `pip install`.
+scripts = ['tardis/scripts/tardis']
+
+setup(scripts=scripts,
+      use_scm_version={'write_to': os.path.join('tardis', 'version.py'),
                        'write_to_template': VERSION_TEMPLATE,
-                       'version_scheme': 'calver-by-date'})
+                       'version_scheme': 'calver-by-date'},)

tardis/scripts/tardis.py → tardis/scripts/tardis

@@ -1,13 +1,25 @@
 #!/usr/bin/env python
-from tardis.io import config_reader
 
-from tardis.simulation import Simulation
+import warnings
+import argparse
+import logging
 import numpy as np
+from tardis.io import config_reader
+from tardis.simulation import Simulation
 
-import logging
-import argparse
 
-import warnings
+def get_virtual_spectrum():
+    # Catch warning when acessing invalid spectrum_virtual
+    with warnings.catch_warnings(record=True) as w:
+        spectrum = simulation.runner.spectrum_virtual
+
+        if len(w) > 0 and w[-1]._category_name == "UserWarning":
+            warnings.warn(
+                "Virtual spectrum is not available, using the "
+                "real packet spectrum instead."
+            )
+            spectrum = simulation.runner.spectrum
+    return spectrum
 
 
 tardis_description = """
@@ -80,44 +92,31 @@
 tardis_config = config_reader.Configuration.from_yaml(args.config_fname)
 simulation = Simulation.from_config(tardis_config)
 
-
-def get_virtual_spectrum():
-    # Catch warning when acessing invalid spectrum_virtual
-    with warnings.catch_warnings(record=True) as w:
-        spectrum = simulation.runner.spectrum_virtual
-
-        if len(w) > 0 and w[-1]._category_name == "UserWarning":
-            warnings.warn(
-                "Virtual spectrum is not available, using the "
-                "real packet spectrum instead."
-            )
-            spectrum = simulation.runner.spectrum
-    return spectrum
-
-
 print("Saving the {} spectrum.".format(tardis_config.spectrum.method))
 
-# Believe it or not, that is a fancy switch-case statement
-# We need functions so that only one spectrum is accessed so we can catch
+# Believe it or not, that is a fancy switch-case statement. We need
+# functions so that only one spectrum is accessed so we can catch
 # the warning properly
+
 get_spectrum = {
     "real": lambda: simulation.runner.spectrum,
     "virtual": get_virtual_spectrum,
     "integrated": lambda: simulation.runner.spectrum_integrated,
 }[tardis_config.spectrum.method]
 
-
 if args.gdb:
     import os
 
     print(os.getpid())
     raw_input()  # Workaround to attach gdb
+
 if args.profile:
     import cProfile
 
     cProfile.runctx(
         "simulation.run()", locals(), globals(), filename=args.profiler_log_file
     )
+
 else:
     simulation.run()