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

Remove aesara.compile.ops.expandable_types

上级 cbf91122
...@@ -213,11 +213,6 @@ class DeepCopyOp(COp): ...@@ -213,11 +213,6 @@ class DeepCopyOp(COp):
deep_copy_op = DeepCopyOp() deep_copy_op = DeepCopyOp()
# List of Aesara Types that one can add an extra dimension and for which
# Scan can deal with.
expandable_types = ()
def load_back(mod, name): def load_back(mod, name):
__import__(mod) __import__(mod)
import sys import sys
......
...@@ -4,7 +4,7 @@ from collections import OrderedDict ...@@ -4,7 +4,7 @@ from collections import OrderedDict
import numpy as np import numpy as np
import aesara.tensor as aet import aesara.tensor as aet
from aesara.compile import SharedVariable, ops from aesara.compile import SharedVariable
from aesara.compile.function import function from aesara.compile.function import function
from aesara.compile.mode import Mode from aesara.compile.mode import Mode
from aesara.configdefaults import config from aesara.configdefaults import config
...@@ -869,7 +869,7 @@ def scan( ...@@ -869,7 +869,7 @@ def scan(
new_var = safe_new(input.variable) new_var = safe_new(input.variable)
if getattr(input.variable, "name", None) is not None: if getattr(input.variable, "name", None) is not None:
new_var.name = input.variable.name + "_copy" new_var.name = input.variable.name + "_copy"
if isinstance(new_var.type, ops.expandable_types): if isinstance(new_var.type, TensorType):
sit_sot_inner_inputs.append(new_var) sit_sot_inner_inputs.append(new_var)
sit_sot_scan_inputs.append( sit_sot_scan_inputs.append(
utils.expand_empty( utils.expand_empty(
......
...@@ -598,9 +598,6 @@ class TensorType(CType): ...@@ -598,9 +598,6 @@ class TensorType(CType):
return np.dtype(self.dtype).itemsize return np.dtype(self.dtype).itemsize
aesara.compile.ops.expandable_types += (TensorType,)
def values_eq_approx( def values_eq_approx(
a, b, allow_remove_inf=False, allow_remove_nan=False, rtol=None, atol=None a, b, allow_remove_inf=False, allow_remove_nan=False, rtol=None, atol=None
): ):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论