ob-metaflow 2.10.1.6__py2.py3-none-any.whl → 2.10.2.6__py2.py3-none-any.whl

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

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

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.dist-info → ob_metaflow-2.10.2.6.dist-info}/METADATA

@@ -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.dist-info → ob_metaflow-2.10.2.6.dist-info}/RECORD

@@ -144,10 +144,10 @@ metaflow/plugins/airflow/sensors/base_sensor.py,sha256=lgbd9JjfDBjDRT8-oEnPnFNst
 metaflow/plugins/airflow/sensors/external_task_sensor.py,sha256=GAWfc2GJqSSAKOJyHSdMea1-tu1cLgEWlRIfe3fti3Y,6048
 metaflow/plugins/airflow/sensors/s3_sensor.py,sha256=zym4mUm_f_gBsvHHVqGtX_OOxRjM3WG217gaT-XDBnk,3274
 metaflow/plugins/argo/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-metaflow/plugins/argo/argo_client.py,sha256=hpUKrYDFTRO4TqMy6OX4u6Vy6xUOjOXzzC9OSAxcbQs,16346
+metaflow/plugins/argo/argo_client.py,sha256=lkVfUuH4rT1ONBvPRWZwQoMDUphHitFxCYWdXRU3zZY,15855
 metaflow/plugins/argo/argo_events.py,sha256=Pt5L7DhxxWMES81JS8kCh4Al4w7k-uKeH99FChBPp1o,6002
-metaflow/plugins/argo/argo_workflows.py,sha256=cp1hEUSzRut6Zeyg6IcVOFEDpba1ZHny4mdjiqmg73w,115971
-metaflow/plugins/argo/argo_workflows_cli.py,sha256=hP2trNZoiw0oGUVDsIr7hxUzdPltHXpvpJasYa2xyOc,32334
+metaflow/plugins/argo/argo_workflows.py,sha256=tr00xdYl_TTT8J0E1q3SnNpREAClyz7ZJIcTjhyRKlw,116131
+metaflow/plugins/argo/argo_workflows_cli.py,sha256=oXCSU6TdkF-ptQTXux_W2KAwL-LMJjpMvoxLG9Et0o8,32350
 metaflow/plugins/argo/argo_workflows_decorator.py,sha256=CfKVoHCOsCCQMghhPE30xw15gacwp3hR23HCo9ZZFVg,6580
 metaflow/plugins/argo/process_input_paths.py,sha256=4SiUoxbnTX4rCt0RSLcxG5jysbyd8oU-5JT0UOgy-vk,555
 metaflow/plugins/aws/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -289,9 +289,9 @@ metaflow/tutorials/07-worldview/README.md,sha256=5vQTrFqulJ7rWN6r20dhot9lI2sVj9W
 metaflow/tutorials/07-worldview/worldview.ipynb,sha256=ztPZPI9BXxvW1QdS2Tfe7LBuVzvFvv0AToDnsDJhLdE,2237
 metaflow/tutorials/08-autopilot/README.md,sha256=GnePFp_q76jPs991lMUqfIIh5zSorIeWznyiUxzeUVE,1039
 metaflow/tutorials/08-autopilot/autopilot.ipynb,sha256=DQoJlILV7Mq9vfPBGW-QV_kNhWPjS5n6SJLqePjFYLY,3191
-ob_metaflow-2.10.1.6.dist-info/LICENSE,sha256=nl_Lt5v9VvJ-5lWJDT4ddKAG-VZ-2IaLmbzpgYDz2hU,11343
-ob_metaflow-2.10.1.6.dist-info/METADATA,sha256=rRy0R2xJQWdXHgwR7Ud4641M-SswA12GdPz7PP8VWIM,5061
-ob_metaflow-2.10.1.6.dist-info/WHEEL,sha256=iYlv5fX357PQyRT2o6tw1bN-YcKFFHKqB_LwHO5wP-g,110
-ob_metaflow-2.10.1.6.dist-info/entry_points.txt,sha256=IKwTN1T3I5eJL3uo_vnkyxVffcgnRdFbKwlghZfn27k,57
-ob_metaflow-2.10.1.6.dist-info/top_level.txt,sha256=v1pDHoWaSaKeuc5fKTRSfsXCKSdW1zvNVmvA-i0if3o,9
-ob_metaflow-2.10.1.6.dist-info/RECORD,,
+ob_metaflow-2.10.2.6.dist-info/LICENSE,sha256=nl_Lt5v9VvJ-5lWJDT4ddKAG-VZ-2IaLmbzpgYDz2hU,11343
+ob_metaflow-2.10.2.6.dist-info/METADATA,sha256=wG48Bi6r3Exgd1JZcPs3N79aZSu6FjjuajiQfAMBQ0k,5061
+ob_metaflow-2.10.2.6.dist-info/WHEEL,sha256=iYlv5fX357PQyRT2o6tw1bN-YcKFFHKqB_LwHO5wP-g,110
+ob_metaflow-2.10.2.6.dist-info/entry_points.txt,sha256=IKwTN1T3I5eJL3uo_vnkyxVffcgnRdFbKwlghZfn27k,57
+ob_metaflow-2.10.2.6.dist-info/top_level.txt,sha256=v1pDHoWaSaKeuc5fKTRSfsXCKSdW1zvNVmvA-i0if3o,9
+ob_metaflow-2.10.2.6.dist-info/RECORD,,