override spec parser

metaflow/plugins/kubernetes/kubernetes_decorator.py

@@ -400,6 +400,22 @@ def _save_package_once(cls, flow_datastore, package):
  [package.blob], len_hint=1
  )[0]
 
+ @classmethod
+ def _parse_decorator_spec(cls, deco_spec: str):
+ if not deco_spec:
+ return cls()
+
+ valid_options = "|".join(cls.defaults.keys())
+ deco_spec_parts = []
+ for part in re.split(f""",(?=[\s\w]+[{valid_options}]=)""", deco_spec):
+ name, val = part.split("=", 1)
+ if name in {"labels", "node_selector"}:
+ both = name == "node_selector"
+ val = json.dumps(cls.parse_kube_keyvalue_list(val.split(","), both))
+ deco_spec_parts.append("=".join([name, val]))
+ deco_spec_parsed = ",".join(deco_spec_parts)
+ return super()._parse_decorator_spec(deco_spec_parsed)
+
  @staticmethod
  def parse_kube_keyvalue_list(items: List[str], requires_both: bool = True):
  try: