From 3bc51f949a33cadac24006ec422af83713ded17f Mon Sep 17 00:00:00 2001 From: igalshilman Date: Thu, 18 Jul 2024 09:16:08 +0000 Subject: [PATCH] Add additional imports to __init__.py --- examples/example.py | 8 +++----- python/restate/__init__.py | 9 +++++++++ 2 files changed, 12 insertions(+), 5 deletions(-) diff --git a/examples/example.py b/examples/example.py index fd68573..c6702e9 100644 --- a/examples/example.py +++ b/examples/example.py @@ -14,11 +14,9 @@ from datetime import timedelta -from restate.service import Service -from restate.context import Context, ObjectContext, ObjectSharedContext -from restate.context import WorkflowContext, WorkflowSharedContext -from restate.object import VirtualObject -from restate.workflow import Workflow +from restate import Service, Context +from restate import VirtualObject, ObjectContext, ObjectSharedContext +from restate import Workflow, WorkflowContext, WorkflowSharedContext import restate greeter = Service("greeter") diff --git a/python/restate/__init__.py b/python/restate/__init__.py index 956a73f..5acb1bd 100644 --- a/python/restate/__init__.py +++ b/python/restate/__init__.py @@ -12,4 +12,13 @@ Restate SDK for Python """ +from .service import Service +from .object import VirtualObject +from .workflow import Workflow + +# types +from .context import Context, ObjectContext, ObjectSharedContext +from .context import WorkflowContext, WorkflowSharedContext +from .context import DurablePromise + from .endpoint import app