From a1cbd001b8de4b38e2fa345c8892eb09248699c4 Mon Sep 17 00:00:00 2001 From: braniii Date: Thu, 5 May 2022 18:09:14 +0200 Subject: [PATCH] Fix black issue of CircleCI --- sklearn_extra/kernel_methods/_eigenpro.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sklearn_extra/kernel_methods/_eigenpro.py b/sklearn_extra/kernel_methods/_eigenpro.py index b7e0b060..7c6f345f 100644 --- a/sklearn_extra/kernel_methods/_eigenpro.py +++ b/sklearn_extra/kernel_methods/_eigenpro.py @@ -180,7 +180,7 @@ def _setup(self, feat, max_components, mG, alpha): ] max_S = E[0].astype(np.float32) - kxx = 1 - np.sum(Lambda ** 2, axis=1) * n_subsamples + kxx = 1 - np.sum(Lambda**2, axis=1) * n_subsamples return max_S / scale, np.max(kxx), D, Lambda def _initialize_params(self, X, Y, random_state): @@ -234,7 +234,7 @@ def _initialize_params(self, X, Y, random_state): n_components = max(1, n_components) # Approximate amount of memory that we want to use - mem_bytes = 0.1 * 1024 ** 3 + mem_bytes = 0.1 * 1024**3 # Memory used with a certain sample size mem_usages = (d + n_label + 2 * np.arange(sample_size)) * n * 4 mG = np.int32(np.sum(mem_usages < mem_bytes))