add missing parameters for docsig

basf · Aug 28, 2024 · dd0ebbe · dd0ebbe
1 parent 3dbcff8
commit dd0ebbe
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/test_extras/test_chemprop/test_chemprop_pipeline.py b/test_extras/test_chemprop/test_chemprop_pipeline.py
@@ -143,6 +143,11 @@ def get_classification_pipeline() -> Pipeline:
 def get_multiclass_classification_pipeline(n_classes: int) -> Pipeline:
     """Get the Chemprop model pipeline for classification.
 
+    Parameters
+    ----------
+    n_classes : int
+        The number of classes for model initialization.
+
     Returns
     -------
     Pipeline