improve docstring

basf · Jun 20, 2024 · a46ed61 · a46ed61
1 parent 33dcc70
commit a46ed61
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/molpipeline/estimators/chemprop/loss_wrapper.py b/molpipeline/estimators/chemprop/loss_wrapper.py
@@ -41,6 +41,10 @@ def get_params(self: _LossFunction, deep: bool = True) -> dict[str, Any]:
         deep : bool, optional
             Not used, only present to match the sklearn API.
 
+        Returns
+        -------
+        dict[str, Any]
+            The parameters of the loss function.
         """
         return {"task_weights": self._original_task_weights}