document return value for method LearnerDF.fit_predict()

BCG-X-Official · Aug 26, 2020 · e1edc34 · e1edc34
1 parent e4e68b2
commit e1edc34
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/sklearndf/_sklearndf.py b/src/sklearndf/_sklearndf.py
@@ -191,6 +191,8 @@ def fit_predict(
         :param y: a series or data frame with one or more outputs per observation
         :param fit_params: optional keyword parameters as required by specific \
             learner implementations
+        :return predictions per observation as a series, or as a data frame in case of \
+            multiple outputs
         """
         pass