From ed538e5d056ef591471a0661f54bf723af7d50e2 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Mon, 18 Dec 2023 21:03:12 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- src/pyepal/models/coregionalized.py | 1 - src/pyepal/pal/pal_gpy.py | 1 - src/pyepal/pal/pal_neural_tangent.py | 2 +- 3 files changed, 1 insertion(+), 3 deletions(-) diff --git a/src/pyepal/models/coregionalized.py b/src/pyepal/models/coregionalized.py index 77607b2..58c1969 100644 --- a/src/pyepal/models/coregionalized.py +++ b/src/pyepal/models/coregionalized.py @@ -43,7 +43,6 @@ def __init__( # pylint:disable=too-many-arguments W_rank=1, kernel_name="coreg", ): - # Input and Output ( X, # pylint:disable=invalid-name diff --git a/src/pyepal/pal/pal_gpy.py b/src/pyepal/pal/pal_gpy.py index ec197b6..297b9a4 100644 --- a/src/pyepal/pal/pal_gpy.py +++ b/src/pyepal/pal/pal_gpy.py @@ -91,7 +91,6 @@ def _train(self): pass # There is no training in instance based models def _predict(self): - from ..models.gpr import predict # pylint:disable=import-outside-toplevel means, stds = [], [] diff --git a/src/pyepal/pal/pal_neural_tangent.py b/src/pyepal/pal/pal_neural_tangent.py index 61ed564..3652e28 100644 --- a/src/pyepal/pal/pal_neural_tangent.py +++ b/src/pyepal/pal/pal_neural_tangent.py @@ -37,6 +37,7 @@ __all__ = ["PALNT", "NTModel"] + # We move those functions out of the class so that we can parallelize them def _set_one_infinite_width_model( # pylint:disable=too-many-arguments i: int, @@ -127,7 +128,6 @@ def __init__(self, *args, **kwargs): def _set_data(self): self.design_space = self.design_space_scaler.fit_transform(self.design_space) for i, _ in enumerate(self.models): - predict_fn, scaler = _set_one_infinite_width_model( i, self.models,