Skip to content

Commit c658199

Browse files
committed
clean up/ simplify/move import to top
1 parent ab6c74f commit c658199

2 files changed

Lines changed: 5 additions & 17 deletions

File tree

pvlib/iam.py

Lines changed: 4 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
import functools
1414
from scipy.optimize import minimize
1515
from pvlib.tools import cosd, sind, acosd
16+
from scipy.interpolate import make_interp_spline
1617

1718
# a dict of required parameter names for each IAM model
1819
# keys are the function names for the IAM models
@@ -469,9 +470,6 @@ def interp(aoi, theta_ref, iam_ref, method='linear', normalize=True):
469470
pvlib.iam.sapm
470471
'''
471472
# Contributed by Anton Driesse (@adriesse), PV Performance Labs. July, 2019
472-
473-
from scipy.interpolate import make_interp_spline
474-
475473
# Scipy doesn't give the clearest feedback, so check number of points here.
476474
MIN_REF_VALS = {'linear': 2, 'quadratic': 3, 'cubic': 4, 1: 2, 2: 3, 3: 4}
477475

@@ -487,22 +485,13 @@ def interp(aoi, theta_ref, iam_ref, method='linear', normalize=True):
487485
iam_ref = np.asarray(iam_ref)
488486

489487
if method == "linear":
490-
spline = make_interp_spline(theta_ref, iam_ref, k=1)
491-
492-
def interpolator(x):
493-
return spline(x)
488+
interpolator = make_interp_spline(theta_ref, iam_ref, k=1)
494489

495490
elif method == "quadratic":
496-
spline = make_interp_spline(theta_ref, iam_ref, k=2)
497-
498-
def interpolator(x):
499-
return spline(x)
491+
interpolator = make_interp_spline(theta_ref, iam_ref, k=2)
500492

501493
elif method == "cubic":
502-
spline = make_interp_spline(theta_ref, iam_ref, k=3)
503-
504-
def interpolator(x):
505-
return spline(x)
494+
interpolator = make_interp_spline(theta_ref, iam_ref, k=3)
506495

507496
else:
508497
raise ValueError(f"Invalid interpolation method '{method}'.")

pvlib/spectrum/response.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -68,8 +68,7 @@ def get_example_spectral_response(wavelength=None):
6868
wavelength = np.arange(280, 1200 + resolution, resolution)
6969
x = SR_DATA[0]
7070
y = SR_DATA[1]
71-
spline = make_interp_spline(
72-
x, y, k=3)
71+
spline = make_interp_spline(x, y, k=3)
7372

7473
values = spline(wavelength)
7574
values[(wavelength < x[0]) | (wavelength > x[-1])] = 0.0

0 commit comments

Comments
 (0)