Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Oct 2, 2023
1 parent f446b3b commit c9f4d75
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion src/stream_ml/pytorch/_connect/data.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ def _from_tensor_to_ndarray(
return replace(data, array=np.asarray(data.array, **kwargs))


ASTYPE_REGISTRY[(xp.Tensor, np.ndarray)] = _from_tensor_to_ndarray # type: ignore[assignment] # noqa: E501
ASTYPE_REGISTRY[(xp.Tensor, np.ndarray)] = _from_tensor_to_ndarray # type: ignore[assignment]


try:
Expand Down
2 changes: 1 addition & 1 deletion src/stream_ml/pytorch/_connect/scaler.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,4 +29,4 @@ def standard_scaler_astype_tensor(
)


ASTYPE_REGISTRY[(StandardScaler, xp.Tensor)] = standard_scaler_astype_tensor # type: ignore[assignment] # noqa: E501
ASTYPE_REGISTRY[(StandardScaler, xp.Tensor)] = standard_scaler_astype_tensor # type: ignore[assignment]
2 changes: 1 addition & 1 deletion src/stream_ml/pytorch/prior/_track.py
Original file line number Diff line number Diff line change
Expand Up @@ -131,7 +131,7 @@ def logpdf(
The logpdf.
"""
# Get model parameters evaluated at the control points. shape (C, 1).
cmpars = model.unpack_params(model(self._x)) # type: ignore[call-overload] # noqa: E501
cmpars = model.unpack_params(model(self._x)) # type: ignore[call-overload]
cmp_arr = xp.stack( # (C, F)
tuple(cmpars[(n, self.component_param_name)] for n in self._y_names), 1
)
Expand Down

0 comments on commit c9f4d75

Please sign in to comment.