add main entry point for package

arcalot · Dec 18, 2023 · c3f7b6b · c3f7b6b
1 parent f032c59
commit c3f7b6b
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 1 deletion.
diff --git a/arcaflow_plugin_template_python/__main__.py b/arcaflow_plugin_template_python/__main__.py
@@ -0,0 +1,14 @@
+import sys
+from arcaflow_plugin_template_python import plugin
+from arcaflow_plugin_template_python import hello_world
+
+if __name__ == "__main__":
+    sys.exit(
+        plugin.run(
+            plugin.build_schema(
+                # List your step functions here:
+                hello_world,
+            )
+        )
+    )
+
diff --git a/arcaflow_plugin_template_python/arcaflow_plugin_template_python.py b/arcaflow_plugin_template_python/arcaflow_plugin_template_python.py
@@ -3,7 +3,7 @@
 import sys
 import typing
 from arcaflow_plugin_sdk import plugin
-from arcaflow_schema_template_python.arcaflow_schema_template_python import (
+from arcaflow_schema_template_python import (
     InputParams,
     SuccessOutput,
     ErrorOutput,

diff --git a/...python/arcaflow_schema_template_python.py → ...python/arcaflow_schema_template_python.py b/...python/arcaflow_schema_template_python.py → ...python/arcaflow_schema_template_python.py