Commit 1d7902a4 authored by nd-02110114's avatar nd-02110114
Browse files

🚨 fix doctest

parent 2e7f39bc
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -68,7 +68,7 @@ def normalize_weight_shape(w: np.ndarray, n_samples: int,
  Examples
  --------
  >>> import numpy as np
  >>> w_out = dc.metrics.normalize_weight_shape(None, n_samples, n_tasks)
  >>> w_out = normalize_weight_shape(None, n_samples, n_tasks)
  >>> (w_out == np.ones((n_samples, n_tasks))).all()
  True