Fix console script

ubccr · May 20, 2014 · 3ab9e11 · 3ab9e11
1 parent 14edb16
commit 3ab9e11
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/cockatoo/cli.py b/cockatoo/cli.py
@@ -125,11 +125,14 @@ def hclust(ctx, screen, pdist, dendrogram, newick, testing, basename, cutoff, we
     s = cockatoo.screen.parse_json(screen)
     cockatoo.hclust.cluster(s, weights, cutoff, basename, pdist, dendrogram, newick, testing)
 
-if __name__ == '__main__':
+def main():
     logging.basicConfig(
         format='%(asctime)s [%(levelname)s] %(message)s',
         datefmt='%Y-%m-%d %H:%M:%S',
         level=logging.CRITICAL
     )
 
     cli(obj={})
+
+if __name__ == '__main__':
+    main()
diff --git a/setup.py b/setup.py
@@ -21,7 +21,7 @@
     ],
     entry_points='''
         [console_scripts]
-        cockatoo=cockatoo.cli:cli
+        cockatoo=cockatoo.cli:main
     ''',
     classifiers=[
         'Development Status :: 3 - Alpha',