From c826b68523cd922dd0ee1ddfb5f044440d4ddcb7 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 4b3c08ea..29f7715b 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.")