metaflow 2.12.28__py2.py3-none-any.whl → 2.12.29__py2.py3-none-any.whl

metaflow/__init__.py

@@ -101,9 +101,7 @@ from .metaflow_current import current
 # 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 @@ if sys.version_info >= (3, 7):
     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__ = []

metaflow/cli.py

@@ -282,21 +282,31 @@ def dump(obj, input_path, private=None, max_value_size=None, include=None, file=
     else:
         ds_list = list(datastore_set)  # get all tasks
 
+    tasks_processed = False
     for ds in ds_list:
-        echo(
-            "Dumping output of run_id=*{run_id}* "
-            "step=*{step}* task_id=*{task_id}*".format(
-                run_id=ds.run_id, step=ds.step_name, task_id=ds.task_id
-            ),
-            fg="magenta",
-        )
-
-        if file is None:
-            echo_always(
-                ds.format(**kwargs), highlight="green", highlight_bold=False, err=False
+        if ds is not None:
+            tasks_processed = True
+            echo(
+                "Dumping output of run_id=*{run_id}* "
+                "step=*{step}* task_id=*{task_id}*".format(
+                    run_id=ds.run_id, step=ds.step_name, task_id=ds.task_id
+                ),
+                fg="magenta",
             )
-        else:
-            output[ds.pathspec] = ds.to_dict(**kwargs)
+
+            if file is None:
+                echo_always(
+                    ds.format(**kwargs),
+                    highlight="green",
+                    highlight_bold=False,
+                    err=False,
+                )
+            else:
+                output[ds.pathspec] = ds.to_dict(**kwargs)
+
+    if not tasks_processed:
+        echo(f"No task(s) found for pathspec {input_path}", fg="red")
+        return
 
     if file is not None:
         with open(file, "wb") as f:

metaflow/client/core.py

@@ -39,7 +39,7 @@ from ..info_file import INFO_FILE
 from .filecache import FileCache
 
 if TYPE_CHECKING:
-    from metaflow.metadata import MetadataProvider
+    from metaflow.metadata_provider import MetadataProvider
 
 try:
     # python2
@@ -143,7 +143,7 @@ def default_metadata() -> str:
     if default:
         current_metadata = default[0]
     else:
-        from metaflow.plugins.metadata import LocalMetadataProvider
+        from metaflow.plugins.metadata_providers import LocalMetadataProvider
 
         current_metadata = LocalMetadataProvider
     return get_metadata()

metaflow/clone_util.py

@@ -1,5 +1,5 @@
 import time
-from .metadata import MetaDatum
+from .metadata_provider import MetaDatum
 
 
 def clone_task_helper(