doc: add missing type hints

kili-technology · Oct 9, 2023 · 094735e · 094735e
1 parent 35d3a4d
commit 094735e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/kili/entrypoints/mutations/label/__init__.py b/src/kili/entrypoints/mutations/label/__init__.py
@@ -131,7 +131,7 @@ def append_to_labels(
         label_type: LabelType = "DEFAULT",
         project_id: Optional[str] = None,
         seconds_to_label: Optional[int] = 0,
-    ):
+    ) -> Dict[Literal["id"], str]:
         """!!! danger "[DEPRECATED]"
             append_to_labels method is deprecated. Please use append_labels instead.
             This new function allows to import several labels 10 times faster.

diff --git a/src/kili/entrypoints/queries/plugins/__init__.py b/src/kili/entrypoints/queries/plugins/__init__.py
@@ -2,7 +2,7 @@
 
 import json
 from datetime import datetime
-from typing import Optional
+from typing import Dict, List, Optional
 
 from typeguard import typechecked
 
@@ -95,7 +95,7 @@ def get_plugin_status(
         self,
         plugin_name: str,
         verbose: bool = True,
-    ):
+    ) -> str:
         """Update a plugin with new code.
 
         Args:
@@ -116,7 +116,7 @@ def get_plugin_status(
     def list_plugins(
         self,
         fields: ListOrTuple[str] = ("name", "projectIds", "id", "createdAt", "updatedAt"),
-    ):
+    ) -> List[Dict]:
         # pylint: disable=line-too-long
         """List all plugins from your organization.