From 98009c1dc5b8ee4c630bbe994ccbc7c22eef63cb Mon Sep 17 00:00:00 2001 From: Felix Dangel Date: Tue, 18 Jul 2023 16:42:38 -0400 Subject: [PATCH] [FIX] Documentation and rename --- curvlinops/_base.py | 2 +- curvlinops/jacobian.py | 3 --- 2 files changed, 1 insertion(+), 4 deletions(-) diff --git a/curvlinops/_base.py b/curvlinops/_base.py index 7f332ee..e655eb9 100644 --- a/curvlinops/_base.py +++ b/curvlinops/_base.py @@ -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 diff --git a/curvlinops/jacobian.py b/curvlinops/jacobian.py index 58b18e6..65d4614 100644 --- a/curvlinops/jacobian.py +++ b/curvlinops/jacobian.py @@ -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]