Skip to content

Commit 1153dce

Browse files
authored
Merge pull request #1186 from rhayes777/feature/lazy-imports
perf: defer scipy.stats import in PriorVectorized
2 parents 730a133 + 54c62ee commit 1153dce

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

autofit/mapper/prior/vectorized.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
11
import numpy as np
2-
from scipy.stats import norm
32

43
from autofit.mapper.prior.gaussian import GaussianPrior
54
from autofit.mapper.prior.truncated_gaussian import TruncatedGaussianPrior
@@ -97,6 +96,8 @@ def __init__(self, model: AbstractPriorModel):
9796
uppers - self.truncated_gaussian_means
9897
) / self.truncated_gaussian_sigmas
9998

99+
from scipy.stats import norm
100+
100101
self.truncated_gaussian_cdf_a = norm.cdf(a)
101102
self.truncated_gaussian_cdf_b = norm.cdf(b)
102103

@@ -153,6 +154,8 @@ def __call__(self, cube: np.ndarray) -> np.ndarray:
153154

154155
# 3) Batch‐process all GaussianPriors
155156
if self.gaussian_idx:
157+
from scipy.stats import norm
158+
156159
subcube = cube[:, self.gaussian_idx] # (n_samples, n_gaussians)
157160

158161
inv = norm.ppf(subcube) # inverse CDF of standard normal
@@ -163,6 +166,8 @@ def __call__(self, cube: np.ndarray) -> np.ndarray:
163166

164167
subcube = cube[:, self.truncated_gaussian_idx] # (n_samples, n_truncs)
165168

169+
from scipy.stats import norm
170+
166171
truncated_cdf = self.truncated_gaussian_cdf_a + subcube * (
167172
self.truncated_gaussian_cdf_b - self.truncated_gaussian_cdf_a
168173
)

0 commit comments

Comments
 (0)