Merge pull request #4 from iwr-redmond/main

Add CLI entry point (#3)

Author: cmdr2

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "torchruntime"
-version = "1.4.1"
+version = "1.4.2"
 description = "Meant for app developers. A convenient way to install and configure the appropriate version of PyTorch on the user's computer, based on the OS and GPU manufacturer and model number."
 readme = "README.md"
 requires-python = ">=3.0"
@@ -15,6 +15,9 @@ dynamic = [ "dependencies",]
 name = "cmdr2"
 email = "[email protected]"
 
+[project.scripts]
+torchruntime = "torchruntime.__main__:main"
+
 [project.urls]
 Homepage = "https://github.com/easydiffusion/torchruntime"
 "Bug Tracker" = "https://github.com/easydiffusion/torchruntime/issues"

torchruntime/__main__.py

@@ -1,20 +1,20 @@
 from .installer import install
 
 
-def print_usage():
+def print_usage(entry_command: str):
     """Print usage information with examples."""
     usage = """
-Usage: python -m torchruntime <command> [arguments]
+Usage: {entry_command} <command> [arguments]
 
 Commands:
     install             Install PyTorch packages
     --help             Show this help message
 
 Examples:
-    python -m torchruntime install
-    python -m torchruntime install torch==2.2.0 torchvision==0.17.0
-    python -m torchruntime install torch>=2.0.0 torchaudio
-    python -m torchruntime install torch==2.1.* torchvision>=0.16.0 torchaudio==2.1.0
+    {entry_command} install
+    {entry_command} install torch==2.2.0 torchvision==0.17.0
+    {entry_command} install torch>=2.0.0 torchaudio
+    {entry_command} install torch==2.1.* torchvision>=0.16.0 torchaudio==2.1.0
 
 If no packages are specified, the latest available versions
 of torch, torchaudio and torchvision will be installed.
@@ -26,15 +26,21 @@ def print_usage():
     package~=2.1.0     Compatible release
     package==2.1.*     Any 2.1.x version
     package            Latest version
-    """
+    """.format(entry_command=entry_command)
     print(usage.strip())
 
 
 def main():
     import sys
 
     if len(sys.argv) < 2 or sys.argv[1] in ["--help", "-h"]:
-        print_usage()
+        # get the entry point
+        entry_path = sys.argv[0]
+        if "__main__.py" in entry_path:
+            cli = "python -m torchruntime"
+        else:
+            cli = "torchruntime"
+        print_usage(cli)
         return
 
     command = sys.argv[1]