simplify wyvern import - move top priority import to wyvern/__init__.py

Skyvern-AI · Aug 24, 2023 · 168e6b7 · 168e6b7
1 parent 327c1e6
commit 168e6b7
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "wyvern-ai"
-version = "0.0.10"
+version = "0.0.11"
 description = ""
 authors = ["Wyvern AI <info@wyvern.ai>"]
 readme = "README.md"

diff --git a/wyvern/__init__.py b/wyvern/__init__.py
@@ -1,6 +1,12 @@
 # -*- coding: utf-8 -*-
+from wyvern.components.pipeline_component import PipelineComponent
 from wyvern.wyvern_logging import setup_logging
 from wyvern.wyvern_tracing import setup_tracing
 
 setup_logging()
 setup_tracing()
+
+
+__all__ = [
+    "PipelineComponent",
+]