From f43971f505e919a03ea2df182211daec79dda64b Mon Sep 17 00:00:00 2001 From: Steffen Schneider Date: Sun, 27 Oct 2024 19:02:10 +0100 Subject: [PATCH] Fix docstring error --- cebra/integrations/sklearn/metrics.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cebra/integrations/sklearn/metrics.py b/cebra/integrations/sklearn/metrics.py index 4b3c08e..29f7715 100644 --- a/cebra/integrations/sklearn/metrics.py +++ b/cebra/integrations/sklearn/metrics.py @@ -206,7 +206,7 @@ def infonce_to_goodness_of_fit(infonce: Union[float, Iterable[float]], Numpy array containing the goodness of fit values, measured in bits Raises: - ``RuntimeError``, if provided model is not fit to data. + RuntimeError: If the provided model is not fit to data. """ if not hasattr(model, "state_dict_"): raise RuntimeError("Fit the CEBRA model first.")