fix a __main__ file to allow profiling and expose the CommandId class

microsoft · Aug 22, 2024 · c020d98 · c020d98
1 parent 687b19f
commit c020d98
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/rats-apps/src/python/rats/cli/__init__.py b/rats-apps/src/python/rats/cli/__init__.py
@@ -1,6 +1,6 @@
 """Uses `rats.cli` to streamline the creation of CLI commands written with Click."""
 
-from ._annotations import command, get_class_commands, get_class_groups, group
+from ._annotations import CommandId, command, get_class_commands, get_class_groups, group
 from ._app import ClickApp
 from ._container import CommandContainer
 from ._plugin import PluginContainer, PluginServices
@@ -13,5 +13,6 @@
     "group",
     "ClickApp",
     "PluginServices",
+    "CommandId",
     "CommandContainer",
 ]
diff --git a/rats-devtools/src/python/rats/devtools/__main__.py b/rats-devtools/src/python/rats/devtools/__main__.py
@@ -1,6 +1,5 @@
 """rats-devtools provides tooling for building commands that help us develop the rats libraries."""
 
-from ._app import run
+from rats.devtools import run
 
-if __name__ == "__main__":
-    run()
+run()