Fix the installed dmypy script (#6029)

Fixes #6028.
python · Dec 7, 2018 · c30dea5 · c30dea5
1 parent 0aa1d5c
commit c30dea5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/mypy/dmypy.py b/mypy/dmypy.py
@@ -492,6 +492,9 @@ def is_running() -> bool:
 
 
 # Run main().
+def console_entry() -> None:
+    main(sys.argv[1:])
+
 
 if __name__ == '__main__':
-    main(sys.argv[1:])
+    console_entry()
diff --git a/setup.py b/setup.py
@@ -157,7 +157,7 @@ def run(self):
       package_data={'mypy': package_data},
       entry_points={'console_scripts': ['mypy=mypy.__main__:console_entry',
                                         'stubgen=mypy.stubgen:main',
-                                        'dmypy=mypy.dmypy:main',
+                                        'dmypy=mypy.dmypy:console_entry',
                                         ]},
       classifiers=classifiers,
       cmdclass=cmdclass,