Skip to content

Commit

Permalink
Fix docstring error
Browse files Browse the repository at this point in the history
  • Loading branch information
stes committed Oct 27, 2024
1 parent 5703628 commit 5e21cdc
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion cebra/integrations/sklearn/metrics.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.")
Expand Down

0 comments on commit 5e21cdc

Please sign in to comment.