Skip to content

Commit

Permalink
[FIX] Documentation and rename
Browse files Browse the repository at this point in the history
  • Loading branch information
f-dangel committed Jul 18, 2023
1 parent a4d668a commit 98009c1
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 4 deletions.
2 changes: 1 addition & 1 deletion curvlinops/_base.py
Original file line number Diff line number Diff line change
Expand Up @@ -79,7 +79,7 @@ def __init__(
self._device = self._infer_device(self._params)
self._progressbar = progressbar

self._num_data = sum(X.shape[0] for (X, _) in self._loop_over_data())
self._N_data = sum(X.shape[0] for (X, _) in self._loop_over_data())

if check_deterministic:
old_device = self._device
Expand Down
3 changes: 0 additions & 3 deletions curvlinops/jacobian.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,9 +49,6 @@ def __init__(
data: Iterable of batched input-target pairs.
progressbar: Show progress bar.
check_deterministic: Check if model and data are deterministic.
Raises:
RuntimeError: If deterministic checks are enables and fail.
"""
num_data = sum(t.shape[0] for t, _ in data)
x = next(iter(data))[0]
Expand Down

0 comments on commit 98009c1

Please sign in to comment.