Skip to content

Commit

Permalink
oops
Browse files Browse the repository at this point in the history
  • Loading branch information
thomaswmorris committed Jul 19, 2023
1 parent edfac57 commit 1091233
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
4 changes: 2 additions & 2 deletions .pre-commit-config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -11,9 +11,9 @@ repos:
rev: 23.1.0
hooks:
- id: black
language_version: python3.10
language_version: python3
- id: black-jupyter
language_version: python3.10
language_version: python3
- repo: https://github.com/pycqa/flake8
rev: 6.0.0
hooks:
Expand Down
1 change: 1 addition & 0 deletions bloptools/bayesian/kernels.py
Original file line number Diff line number Diff line change
Expand Up @@ -68,6 +68,7 @@ def __init__(
raw_diag_entries_initial = (
diag_entries_constraint.inverse_transform(torch.tensor(1e-1))
* torch.ones(self.num_outputs, self.num_inputs).double()
)

self.register_parameter(name="raw_diag_entries", parameter=torch.nn.Parameter(raw_diag_entries_initial))
self.register_constraint(param_name="raw_diag_entries", constraint=diag_entries_constraint)
Expand Down

0 comments on commit 1091233

Please sign in to comment.