fixed a bug in from_tokens documentation

Aleph-Alpha · Feb 21, 2024 · 1188f58 · 1188f58
1 parent 65688cd
commit 1188f58
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/aleph_alpha_client/prompt.py b/aleph_alpha_client/prompt.py
@@ -83,7 +83,7 @@ class Tokens:
             attention for given token ids.
 
     Examples:
-        >>> token_ids = Tokens([1, 2, 3])
+        >>> token_ids = Tokens([1, 2, 3], controls=[])
         >>> prompt = Prompt([token_ids])
     """
 
@@ -470,7 +470,7 @@ def from_tokens(
     ) -> "Prompt":
         """
         Examples:
-            >>> prompt = Prompt.from_tokens(Tokens([1, 2, 3]))
+            >>> prompt = Prompt.from_tokens([1, 2, 3])
         """
         return Prompt([Tokens(tokens, controls or [])])