Fix outdated docstring

huggingface · Sep 18, 2024 · a9bb3f5 · a9bb3f5
1 parent 6a0f3fe
commit a9bb3f5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/setfit/sampler.py b/src/setfit/sampler.py
@@ -44,7 +44,8 @@ def __init__(
         """Generates positive and negative text pairs for contrastive learning.
 
         Args:
-            examples (InputExample): text and labels in a text transformer dataclass
+            sentences (List[str]): text sentences to generate pairs from
+            labels (List[Union[int, float]]): labels for each sentence
             multilabel: set to process "multilabel" labels array
             sampling_strategy: "unique", "oversampling", or "undersampling"
             num_iterations: if provided explicitly sets the number of pairs to be generated