Small fixes

Author: romain-intel

metaflow/plugins/events_decorator.py

@@ -43,7 +43,7 @@ class TriggerDecorator(FlowDecorator):
 
     Parameters
     ----------
-    event : Union[str, Dict[str, Any]], default None
+    event : Union[str, Dict[str, Any]], optional, default None
         Event dependency for this flow.
     events : List[Union[str, Dict[str, Any]]], default []
         Events dependency for this flow.

metaflow/plugins/kubernetes/kubernetes_decorator.py

@@ -71,7 +71,8 @@ class KubernetesDecorator(StepDecorator):
         the scheduled node should not have GPUs.
     gpu_vendor : str, default KUBERNETES_GPU_VENDOR
         The vendor of the GPUs to be used for this step.
-    tolerations : List[str], default METAFLOW_KUBERNETES_TOLERATIONS
+    tolerations : List[str], default []
+        The default is extracted from METAFLOW_KUBERNETES_TOLERATIONS.
         Kubernetes tolerations to use when launching pod in Kubernetes.
     use_tmpfs : bool, default False
         This enables an explicit tmpfs mount for this step.

stubs/setup.py

@@ -1,6 +1,6 @@
 from setuptools import find_packages, setup
 
-version = "2.10.2"
+version = "2.10.8"
 
 setup(
     name="metaflow-stubs",

stubs/stub_generator.py

@@ -605,7 +605,7 @@ def exploit_default(default_value: Any) -> Optional[str]:
         buff.write(")" + ret_annotation + ":\n")
 
         if doc is not None:
-            buff.write('%s%s"""\n' % (indentation, TAB))
+            buff.write('%s%sr"""\n' % (indentation, TAB))
             doc = cast(str, deindent_docstring(doc))
             init_blank = True
             for line in doc.split("\n"):
@@ -705,9 +705,14 @@ def write_out(self):
                     + "\n\n"
                 )
                 f.write("from __future__ import annotations\n\n")
+                imported_typing = False
                 for module in self._imports:
                     f.write("import " + module + "\n")
+                    if module == "typing":
+                        imported_typing = True
                 if self._typing_imports:
+                    if not imported_typing:
+                        f.write("import typing\n")
                     f.write("if typing.TYPE_CHECKING:\n")
                     for module in self._typing_imports:
                         f.write(TAB + "import " + module + "\n")