ENH: Add fitlins.__main__ to allow python -m fitlins

poldracklab · Aug 5, 2020 · 069a01b · 069a01b
1 parent 1e2235c
commit 069a01b
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/fitlins/__main__.py b/fitlins/__main__.py
@@ -0,0 +1,9 @@
+from .cli.run import main
+
+if __name__ == '__main__':
+    import sys
+    from . import __name__
+    # `python -m <module>` typically displays the command as __main__.py
+    if '__main__.py' in sys.argv[0]:
+        sys.argv[0] = '%s -m %s' % (sys.executable, __name__)
+    main()