提交 27423150 authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Fix typing issues in aesara.scalar.math

上级 ed62da3e
...@@ -202,13 +202,14 @@ class Erfinv(UnaryScalarOp): ...@@ -202,13 +202,14 @@ class Erfinv(UnaryScalarOp):
) )
return (gz * cst * exp(erfinv(x) ** 2),) return (gz * cst * exp(erfinv(x) ** 2),)
def c_code(self, node, name, inp, out, sub):
# TODO: erfinv() is not provided by the C standard library # TODO: erfinv() is not provided by the C standard library
# def c_code(self, node, name, inp, out, sub):
# x, = inp # x, = inp
# z, = out # z, = out
# if node.inputs[0].type in complex_types: # if node.inputs[0].type in complex_types:
# raise NotImplementedError('type not supported', type) # raise NotImplementedError('type not supported', type)
# return "%(z)s = erfinv(%(x)s);" % locals() # return "%(z)s = erfinv(%(x)s);" % locals()
raise NotImplementedError()
erfinv = Erfinv(upgrade_to_float_no_complex, name="erfinv") erfinv = Erfinv(upgrade_to_float_no_complex, name="erfinv")
...@@ -236,13 +237,14 @@ class Erfcinv(UnaryScalarOp): ...@@ -236,13 +237,14 @@ class Erfcinv(UnaryScalarOp):
) )
return (-gz * cst * exp(erfcinv(x) ** 2),) return (-gz * cst * exp(erfcinv(x) ** 2),)
def c_code(self, node, name, inp, out, sub):
# TODO: erfcinv() is not provided by the C standard library # TODO: erfcinv() is not provided by the C standard library
# def c_code(self, node, name, inp, out, sub):
# x, = inp # x, = inp
# z, = out # z, = out
# if node.inputs[0].type in complex_types: # if node.inputs[0].type in complex_types:
# raise NotImplementedError('type not supported', type) # raise NotImplementedError('type not supported', type)
# return "%(z)s = erfcinv(%(x)s);" % locals() # return "%(z)s = erfcinv(%(x)s);" % locals()
raise NotImplementedError()
erfcinv = Erfcinv(upgrade_to_float_no_complex, name="erfcinv") erfcinv = Erfcinv(upgrade_to_float_no_complex, name="erfcinv")
...@@ -712,6 +714,9 @@ class GammaIncDer(BinaryScalarOp): ...@@ -712,6 +714,9 @@ class GammaIncDer(BinaryScalarOp):
) )
return np.nan return np.nan
def c_code(self, *args, **kwargs):
raise NotImplementedError()
gammainc_der = GammaIncDer(upgrade_to_float, name="gammainc_der") gammainc_der = GammaIncDer(upgrade_to_float, name="gammainc_der")
...@@ -792,6 +797,9 @@ class GammaIncCDer(BinaryScalarOp): ...@@ -792,6 +797,9 @@ class GammaIncCDer(BinaryScalarOp):
def impl(self, k, x): def impl(self, k, x):
return self.st_impl(k, x) return self.st_impl(k, x)
def c_code(self, *args, **kwargs):
raise NotImplementedError()
gammaincc_der = GammaIncCDer(upgrade_to_float, name="gammaincc_der") gammaincc_der = GammaIncCDer(upgrade_to_float, name="gammaincc_der")
...@@ -900,6 +908,9 @@ class Jv(BinaryScalarOp): ...@@ -900,6 +908,9 @@ class Jv(BinaryScalarOp):
gz * (jv(v - 1, x) - jv(v + 1, x)) / 2.0, gz * (jv(v - 1, x) - jv(v + 1, x)) / 2.0,
] ]
def c_code(self, *args, **kwargs):
raise NotImplementedError()
jv = Jv(upgrade_to_float, name="jv") jv = Jv(upgrade_to_float, name="jv")
...@@ -988,6 +999,9 @@ class Iv(BinaryScalarOp): ...@@ -988,6 +999,9 @@ class Iv(BinaryScalarOp):
gz * (iv(v - 1, x) + iv(v + 1, x)) / 2.0, gz * (iv(v - 1, x) + iv(v + 1, x)) / 2.0,
] ]
def c_code(self, *args, **kwargs):
raise NotImplementedError()
iv = Iv(upgrade_to_float, name="iv") iv = Iv(upgrade_to_float, name="iv")
...@@ -1011,6 +1025,9 @@ class I1(UnaryScalarOp): ...@@ -1011,6 +1025,9 @@ class I1(UnaryScalarOp):
(gz,) = grads (gz,) = grads
return [gz * (i0(x) + iv(2, x)) / 2.0] return [gz * (i0(x) + iv(2, x)) / 2.0]
def c_code(self, *args, **kwargs):
raise NotImplementedError()
i1 = I1(upgrade_to_float, name="i1") i1 = I1(upgrade_to_float, name="i1")
...@@ -1034,6 +1051,9 @@ class I0(UnaryScalarOp): ...@@ -1034,6 +1051,9 @@ class I0(UnaryScalarOp):
(gz,) = grads (gz,) = grads
return [gz * i1(x)] return [gz * i1(x)]
def c_code(self, *args, **kwargs):
raise NotImplementedError()
i0 = I0(upgrade_to_float, name="i0") i0 = I0(upgrade_to_float, name="i0")
...@@ -1251,6 +1271,9 @@ class BetaInc(ScalarOp): ...@@ -1251,6 +1271,9 @@ class BetaInc(ScalarOp):
), ),
] ]
def c_code(self, *args, **kwargs):
raise NotImplementedError()
betainc = BetaInc(upgrade_to_float_no_complex, name="betainc") betainc = BetaInc(upgrade_to_float_no_complex, name="betainc")
...@@ -1436,5 +1459,8 @@ class BetaIncDer(ScalarOp): ...@@ -1436,5 +1459,8 @@ class BetaIncDer(ScalarOp):
) )
return np.nan return np.nan
def c_code(self, *args, **kwargs):
raise NotImplementedError()
betainc_der = BetaIncDer(upgrade_to_float_no_complex, name="betainc_der") betainc_der = BetaIncDer(upgrade_to_float_no_complex, name="betainc_der")
...@@ -139,10 +139,6 @@ check_untyped_defs = False ...@@ -139,10 +139,6 @@ check_untyped_defs = False
ignore_errors = True ignore_errors = True
check_untyped_defs = False check_untyped_defs = False
[mypy-aesara.scalar.math]
ignore_errors = True
check_untyped_defs = False
[mypy-aesara.tensor.type] [mypy-aesara.tensor.type]
ignore_errors = True ignore_errors = True
check_untyped_defs = False check_untyped_defs = False
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论