Skip to content

Commit c504f48

Browse files
committed
fix lint format
1 parent ad4347e commit c504f48

2 files changed

Lines changed: 2 additions & 6 deletions

File tree

src/dynaris/core/nonlinear.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -106,9 +106,7 @@ def tree_flatten(self) -> tuple[list[Array], dict[str, object]]:
106106
return leaves, aux
107107

108108
@classmethod
109-
def tree_unflatten(
110-
cls, aux_data: dict[str, object], children: list[Array]
111-
) -> NonlinearSSM:
109+
def tree_unflatten(cls, aux_data: dict[str, object], children: list[Array]) -> NonlinearSSM:
112110
"""Reconstruct from JAX pytree leaves."""
113111
return cls(
114112
transition_fn=aux_data["transition_fn"], # type: ignore[arg-type]

src/dynaris/core/ssm.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -209,9 +209,7 @@ def fit(
209209
**self._filter_kwargs,
210210
)
211211
elif self._filter_name == "hamilton":
212-
self._filter_result = hamilton_filter(
213-
self._model, obs, initial_state=initial_state
214-
)
212+
self._filter_result = hamilton_filter(self._model, obs, initial_state=initial_state)
215213

216214
self._is_fitted = True
217215
return self

0 commit comments

Comments
 (0)