ob-metaflow 2.10.1.6__tar.gz → 2.10.2.6__tar.gz

{ob-metaflow-2.10.1.6/ob_metaflow.egg-info → ob-metaflow-2.10.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ob-metaflow
-Version: 2.10.1.6
+Version: 2.10.2.6
 Summary: Metaflow: More Data Science, Less Engineering
 Author: Netflix, Outerbounds & the Metaflow Community
 Author-email: help@outerbounds.co

{ob-metaflow-2.10.1.6 → ob-metaflow-2.10.2.6}/metaflow/plugins/argo/argo_client.py

@@ -52,28 +52,15 @@ class ArgoClient(object):
                 json.loads(e.body)["message"] if e.body is not None else e.reason
             )
 
-    def get_workflow_templates(self, flow_name=None):
-        """
-        Return a list of workflow template names for a given Flow name.
-        Returns all workflow template names if no name is provided
-        """
+    def get_workflow_templates(self):
         client = self._client.get()
         try:
-            results = client.CustomObjectsApi().list_namespaced_custom_object(
+            return client.CustomObjectsApi().list_namespaced_custom_object(
                 group=self._group,
                 version=self._version,
                 namespace=self._namespace,
                 plural="workflowtemplates",
             )["items"]
-            return [
-                result["metadata"]["name"]
-                for result in results
-                if flow_name is None
-                or flow_name
-                == result["metadata"]
-                .get("annotations", {})
-                .get("metaflow/flow_name", None)
-            ]
         except client.rest.ApiException as e:
             if e.status == 404:
                 return None

{ob-metaflow-2.10.1.6 → ob-metaflow-2.10.2.6}/metaflow/plugins/argo/argo_workflows.py

@@ -187,13 +187,19 @@ class ArgoWorkflows(object):
     def list_templates(flow_name, all=False):
         client = ArgoClient(namespace=KUBERNETES_NAMESPACE)
 
-        search = None if all else flow_name
-        results = client.get_workflow_templates(search)
-        if results is None:
+        templates = client.get_workflow_templates()
+        if templates is None:
             return []
 
-        template_names = [template_name for template_name in results]
-
+        template_names = [
+            template["metadata"]["name"]
+            for template in templates
+            if all
+            or flow_name
+            == template["metadata"]
+            .get("annotations", {})
+            .get("metaflow/flow_name", None)
+        ]
         return template_names
 
     @staticmethod

{ob-metaflow-2.10.1.6 → ob-metaflow-2.10.2.6}/metaflow/plugins/argo/argo_workflows_cli.py

@@ -842,10 +842,10 @@ def terminate(obj, run_id, authorize=None):
     help="list all Argo Workflow Templates (not just limited to this flow)",
 )
 @click.pass_obj
-def list_templates(obj, all=None):
+def list_workflow_templates(obj, all=None):
     templates = ArgoWorkflows.list_templates(obj.flow.name, all)
     for template_name in templates:
-        obj.echo(template_name)
+        obj.echo_always(template_name)
 
 
 def validate_run_id(

{ob-metaflow-2.10.1.6 → ob-metaflow-2.10.2.6/ob_metaflow.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ob-metaflow
-Version: 2.10.1.6
+Version: 2.10.2.6
 Summary: Metaflow: More Data Science, Less Engineering
 Author: Netflix, Outerbounds & the Metaflow Community
 Author-email: help@outerbounds.co

{ob-metaflow-2.10.1.6 → ob-metaflow-2.10.2.6}/setup.py

@@ -1,6 +1,6 @@
 from setuptools import setup, find_packages
 
-version = "2.10.1.6"
+version = "2.10.2.6"
 
 setup(
     include_package_data=True,