Skip to content
Snippets Groups Projects
Commit 6d276b75 authored by Mads M. Pedersen's avatar Mads M. Pedersen
Browse files

fix gradients of ZongGaussianDeficit and CarbajofuertesGaussianDeficit

parent f0ca5ed8
No related branches found
No related tags found
No related merge requests found
from numpy import newaxis as na
from scipy.special import erf
from py_wake.utils.gradients import erf
import numpy as np
from py_wake.deficit_models import DeficitModel
from py_wake.deficit_models.deficit_model import ConvectionDeficitModel
......
......@@ -21,6 +21,8 @@ from scipy.interpolate._cubic import PchipInterpolator as scipy_PchipInterpolato
from itertools import count
from scipy.interpolate import UnivariateSpline as scipy_UnivariateSpline
from scipy.special import erf as scipy_erf
from autograd.scipy.special import erf as autograd_erf
def asarray(x, dtype=None, order=None):
......@@ -304,6 +306,15 @@ class UnivariateSpline(scipy_UnivariateSpline):
return y
def erf(z):
if isinstance(z, ArrayBox):
return autograd_erf(z)
else:
return scipy_erf(z)
# def get_dtype(arg_lst):
# return (float, np.complex128)[any([np.iscomplexobj(v) for v in arg_lst])]
def trapz(y, x, axis=-1):
if isinstance(y, ArrayBox) or isinstance(x, ArrayBox):
x, y = asarray(x), asarray(y)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment