chore: add Session to top-level imports (#455)

Arize-ai · Mar 27, 2023 · 844c56b · 844c56b
1 parent 096e9b2
commit 844c56b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/phoenix/__init__.py b/src/phoenix/__init__.py
@@ -1,5 +1,5 @@
 from .datasets import Dataset, EmbeddingColumnNames, Schema, load_example, ExampleDatasets
-from .session.session import active_session, close_app, launch_app
+from .session import active_session, close_app, launch_app, Session
 
 # module level doc-string
 __doc__ = """

diff --git a/src/phoenix/session/__init__.py b/src/phoenix/session/__init__.py
@@ -0,0 +1,3 @@
+from .session import active_session, close_app, launch_app, Session
+
+__all__ = ["active_session", "close_app", "launch_app", "Session"]