fixes on CI

zapata-engineering · May 23, 2024 · bca32aa · bca32aa
1 parent 43a2546
commit bca32aa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/projects/orquestra-sdk/src/orquestra/sdk/secrets/__init__.py b/projects/orquestra-sdk/src/orquestra/sdk/secrets/__init__.py
@@ -4,7 +4,7 @@
 # THIS FILE IS A SHIM TO REEXPORT SYMBOLS AS PUBLIC API
 # DO NOT PUT ANY LOGIC INTO THIS FILE.
 
-from orquestra.workflow_runtime.secrets import delete, get, list, set
+from orquestra.workflow_shared.secrets import delete, get, list, set
 
 __all__ = [
     "delete",

diff --git a/projects/orquestra-workflow-shared/src/orquestra/workflow_shared/secrets/_api.py b/projects/orquestra-workflow-shared/src/orquestra/workflow_shared/secrets/_api.py
@@ -35,7 +35,7 @@ class Secret(NamedTuple):
     # specific workspace.
     workspace_id: t.Optional[str] = None
 
-    def __reduce__(self) -> str | tuple[t.Any, ...]:
+    def __reduce__(self) -> t.Union[str, tuple[t.Any, ...]]:
         # We need to override the pickling behaviour for Secret
         # This is because we override other dunder methods which cause the normal
         # picling behaviour to fail.