distribution works

tskit-dev · Feb 24, 2020 · 3076c47 · 3076c47
1 parent 4660150
commit 3076c47
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 6 deletions.
diff --git a/setup.py b/setup.py
@@ -9,11 +9,12 @@
     author_email="[email protected]",
     description="Infer node ages from a tree sequence topology",
     long_description=long_description,
+    packages=["tsdate"],
     long_description_content_type="text/markdown",
     url="http://pypi.python.org/pypi/tsdate",
     entry_points={
         'console_scripts': [
-            'tsdate=tsdate.cli:tsdate_main',
+            'tsdate=tsdate.__main__:main',
         ]
     },
     setup_requires=['setuptools_scm'],

diff --git a/tsdate/__init__.py b/tsdate/__init__.py
@@ -20,9 +20,7 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 
-from .date import (  # NOQA
-    date, build_prior_grid
-    )
 
+from .date import date, build_prior_grid # NOQA
 
-from .provenance import __version__  # NOQA
+from .provenance import __version__ # NOQA
diff --git a/tsdate/__main__.py b/tsdate/__main__.py
@@ -1,4 +1,4 @@
-import tsdate.cli as cli
+from . import cli
 
 
 def main():