Skip to content

Commit 797fa61

Browse files
committed
SklearnModelProcessor - return None if no model
1 parent 26f0f4b commit 797fa61

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/ezmsg/learn/process/sklearn.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -157,9 +157,9 @@ def fit(self, X: np.ndarray, y: np.ndarray) -> None:
157157
else:
158158
raise NotImplementedError("Model does not support fit or learn_many")
159159

160-
def _process(self, message: AxisArray) -> AxisArray:
160+
def _process(self, message: AxisArray) -> AxisArray | None:
161161
if self._state.model is None:
162-
raise RuntimeError("Model has not been fit yet. Call `fit()` or `partial_fit()` before processing.")
162+
return None
163163
X = message.data
164164
original_shape = X.shape
165165
n_input = X.shape[message.get_axis_idx("ch")]

0 commit comments

Comments
 (0)