Skip to content

Commit 6b4343e

Browse files
committed
Adapt gof docstrings
1 parent 5a8de38 commit 6b4343e

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

cebra/integrations/sklearn/metrics.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -117,9 +117,7 @@ def goodness_of_fit(model: cebra_sklearn_cebra.CEBRA) -> List[float]:
117117
different batch sizes or different implementations.
118118
119119
Args:
120-
model: The model to evaluate. This can be an instance of either
121-
`cebra_sklearn_cebra.CEBRA` or `cebra_solver.Solver`.
122-
batch_size: Batch size used to train the model.
120+
model: The model to evaluate. This is an instance of `cebra_sklearn_cebra.CEBRA`.
123121
124122
Returns:
125123
A list of float values representing the goodness of fit for the model.

0 commit comments

Comments
 (0)