Added stub-gen for deployer (try 2)

Netflix · Oct 22, 2024 · 4b09e52 · 4b09e52
1 parent b99c1d6
commit 4b09e52
Show file tree

Hide file tree

Showing 21 changed files with 782 additions and 324 deletions.
diff --git a/metaflow/__init__.py b/metaflow/__init__.py
@@ -101,9 +101,7 @@ class and related decorators.
 # Flow spec
 from .flowspec import FlowSpec
 
-from .parameters import Parameter, JSONTypeClass
-
-JSONType = JSONTypeClass()
+from .parameters import Parameter, JSONTypeClass, JSONType
 
 # data layer
 # For historical reasons, we make metaflow.plugins.datatools accessible as
@@ -149,6 +147,7 @@ class and related decorators.
     from .runner.metaflow_runner import Runner
     from .runner.nbrun import NBRunner
     from .runner.deployer import Deployer
+    from .runner.deployer import DeployedFlow
     from .runner.nbdeploy import NBDeployer
 
 __ext_tl_modules__ = []