Skip to content

Commit a1b4cf3

Browse files
stesCeliaBenquet
andauthored
improve err message
Co-authored-by: Célia Benquet <32598028+CeliaBenquet@users.noreply.github.com>
1 parent bc68be7 commit a1b4cf3

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

cebra/integrations/sklearn/metrics.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -235,7 +235,10 @@ def infonce_to_goodness_of_fit(infonce: Union[float, np.ndarray],
235235
num_sessions = 1
236236
else:
237237
if batch_size is None or num_sessions is None:
238-
raise ValueError("batch_size should be provided if model is not provided.")
238+
raise ValueError(
239+
f"batch_size ({batch_size}) and num_sessions ({num_sessions})"
240+
f"should be provided if model is not provided."
241+
)
239242

240243
nats_to_bits = np.log2(np.e)
241244
chance_level = np.log(model.batch_size * num_sessions)

0 commit comments

Comments
 (0)