From 10912335fefd1b2a370c4282e2fccfa87d71bc70 Mon Sep 17 00:00:00 2001 From: Thomas Morris Date: Wed, 19 Jul 2023 14:16:36 -0400 Subject: [PATCH] oops --- .pre-commit-config.yaml | 4 ++-- bloptools/bayesian/kernels.py | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 6de8684..8e6ebed 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -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: diff --git a/bloptools/bayesian/kernels.py b/bloptools/bayesian/kernels.py index 0a461a4..db2d99f 100644 --- a/bloptools/bayesian/kernels.py +++ b/bloptools/bayesian/kernels.py @@ -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)